SrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSKJ r SS K J r S r S r"S S \5rg)zUtilities for Edge ML API.)absolute_import)division)unicode_literalsN)encoding_helper) operations)util) propertiesc0[R"SSUSS9$)z?Converts model[.tflite] style filename to model_edgetpu.tflite.z (\.tflite)?$z_edgetpu.tflitecountresubsources +lib/googlecloudsdk/api_lib/edgeml/edgeml.py_CompileDestinationrs !2F! DDc0[R"SSUSS9$)zGConverts model[/saved_model.(pb|pbtxt)] style filename to model.tflite.z(/saved_model\.(pb|pbtxt))?$z.tfliter r rrs r_ConvertDestinationr"s /F! LLrc<\rSrSrSrS SjrSrS SjrS SjrSr g) EdgeMlClient'z}Client for Edge ML service. Attributes: client: Generated Edge ML API client. messages: Generated Edge ML API messages. NcU=(d [R"5UlU=(d [R"U5UlURR Ul[R"X5Ul gN) rGetClientInstanceclientGetMessagesModulemessagesprojects_models_servicerOperationsClient_operations_client)selfrr s r__init__EdgeMlClient.__init__/sR4D224DK> 6 6v >DMKK//DM(99&KDrcS[RRRR 5-nUR R UR RUR RU/S9S9US9nURRU5nURRU5n[R"UR R[R "U55nU$)aAnalyzes a machine learning model. The AnalyzeResponse will contain model's framework, and for TF models it will also contain Edge TPU compiliability and input/output tensor information. Args: source: str, GCS object URI to the model file or directory to analyze. Returns: AnalyzedResponse on the finish of analyze operation. Raises: LongrunningError: when long running operation fails. projects/ inputUris gcsSource)analyzeModelRequestproject)r VALUEScorer/ GetOrFailr "EdgemlProjectsModelsAnalyzeRequestAnalyzeModelRequest GcsSourcer"Analyzer$WaitForOperationr JsonToMessageAnalyzeModelResponse MessageToJson)r%rr/ analyze_req operationresultresponses rr6EdgeMlClient.Analyze5s J--22::DDFFG--BB MM==mm---A>CCK %%k2I  $ $ 5 5i @F,, **%%f-/H OrcS[RRRR 5-nU(d [ U5nUR RnURRnUR RUR RU/S9S9nUR RUR RUS9S9nUR RU"UUUS9US9nURR!U5n UR"R%U 5n [&R("UR R*[&R,"U 55n X4$)aOptimizes a TFLite model for EdgeTPU. Args: source: str, GCS object URI to the model file to compile. Must be a .tflite file. destination: str, GCS URI to an output tflite object. If not provided, for source filename `model[.tflite]` this will be set to `model_edgetpu.tflite`. Returns: (CompileModelResponse, output object URI) on the finish of compilation. Raises: LongrunningError: when long running operation fails. r)r*r, outputUrigcsDestination)chipType inputConfig outputConfig)compileModelRequestr/)r r0r1r/r2rr CompileModelRequestChipTypeValueValuesEnum EDGE_TPU_V1 InputConfigr5 OutputConfigGcsDestination"EdgemlProjectsModelsCompileRequestr"Compiler$r7rr8CompileModelResponser:) r%r destinationr/compile_req_type chip_type input_config output_config compile_reqr<r=r>s rrPEdgeMlClient.CompileRsJ J--22::DDFFG '/k}}88 88DDI==,,--))VH)=-?LMM..}}33k3J/LM--BB,$&( CK %%k2I  $ $ 5 5i @F,, **%%f-/H   rcS[RRRR 5-nU(d [ U5nUR RUR RU/S9S9nUR RUR RUS9S9nUR RUR RXES9US9nURRU5nURR!U5n["R$"UR R&["R("U55n X4$)aConverts Tensorflow SavedModel to TFLite model. Args: source: str, GCS URI to an input SavedModel archive destination: str, GCS URI to an output tflite object. If not provided, for source filename `model[/saved_model.(pb|pbtxt)]` this will be set to `model.tflite`. Returns: (ConvertModelResponse, output object URI) on the finish of convert operation. Raises: LongrunningError: when long running operation fails. r)r*r,rArC)rFrG)convertModelRequestr/)r r0r1r/r2rr rLr5rMrN"EdgemlProjectsModelsConvertRequestConvertModelRequestr"Convertr$r7rr8ConvertModelResponser:) r%rrRr/rUrV convert_reqr<r=r>s rr]EdgeMlClient.Convert|s1 J--22::DDFFG '/k==,,--))VH)=-?LMM..}}33k3J/LM--BB MM==$>BCK  %%k2I  $ $ 5 5i @F,, **%%f-/H   r)r$r"rr )NNr) __name__ __module__ __qualname____firstlineno____doc__r&r6rPr]__static_attributes__rrrr'sL :(!T$!rr)re __future__rrrrapitools.base.pyrgooglecloudsdk.api_lib.edgemlrrgooglecloudsdk.corer rrobjectrrgrrrms<!&' ,4.*E M y!6y!r