o 2hU@sdZddlZddlZddlZddlmZddlmZddlm Z Gdddej Z e dd d gZ Gd d d e ej ZGd ddeZejfddZdS)z(Utilities for collecting TFLite metrics.N)Text)converter_error_data_pb2)metricsc@seZdZdZdZdZdZdS) Componentz5Enum class defining name of the converter components.PREPARE_TF_MODELCONVERT_TF_TO_TFLITE_MODELOPTIMIZE_TFLITE_MODELN)__name__ __module__ __qualname____doc__rrrr r _/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/lite/python/convert_phase.pyrs rSubComponentItemname componentc@seZdZdZddZeddZeddZedd Z ed e j Z ed e j Z ed e j Zed e j Zede j Zede j Zede j Zede j Zede jZede jZede jZede jZede jZede jZede jZede jZd S) SubComponentzEnum class defining name of the converter subcomponents. This enum only defines the subcomponents in Python, there might be more subcomponents defined in C++. cC|jjSNvaluerselfr r r__str__1szSubComponent.__str__cCrrrrr r rr4zSubComponent.namecCrr)rrrr r rr8rzSubComponent.component UNSPECIFIEDNVALIDATE_INPUTSLOAD_SAVED_MODELFREEZE_SAVED_MODELCONVERT_KERAS_TO_SAVED_MODEL)CONVERT_CONCRETE_FUNCTIONS_TO_SAVED_MODELFREEZE_KERAS_MODELFREEZE_CONCRETE_FUNCTIONOPTIMIZE_TF_MODEL+CONVERT_GRAPHDEF_USING_DEPRECATED_CONVERTERCONVERT_GRAPHDEFCONVERT_SAVED_MODELCONVERT_JAX_HLO#QUANTIZE_USING_DEPRECATED_QUANTIZER CALIBRATEQUANTIZESPARSIFY)r r r r rpropertyrrrrrrrrrrr r!r"r#rr$r%r&r'rr(r)r*r+r r r rr*sf     rcs8eZdZdZfddZdejfddZddZZ S) ConverterErrorz4Raised when an error occurs during model conversion.cs$tt||g|_||dSr)superr-__init__errors_parse_error_message)rmessage __class__r rr/szConverterError.__init__ error_datacCs|j|dSr)r0append)rr5r r r append_errorszConverterError.append_errorcCsNdtjji}|D]\}}||vr$t}||_||_||dSq dS)aIf the message matches a pattern, assigns the associated error code. It is difficult to assign an error code to some errrors in MLIR side, Ex: errors thrown by other components than TFLite or not using mlir::emitError. This function try to detect them by the error message and assign the corresponding error code. Args: message: The error message of this exception. zFailed to functionalize Control Flow V1 ops. Consider using Control Flow V2 ops instead. See https://www.tensorflow.org/api_docs/python/tf/compat/v1/enable_control_flow_v2.N)rConverterErrorData!ERROR_UNSUPPORTED_CONTROL_FLOW_V1items error_message error_coder7)rr2error_code_mappingpatternr<r5r r rr1s  z#ConverterError._parse_error_message) r r r r r/rr8r7r1 __classcell__r r r3rr-~s  r-csvtvrtdtvrtdtjkrjkrtddtjffdd dtffdd fd d }|S) a-The decorator to identify converter component and subcomponent. Args: component: Converter component name. subcomponent: Converter subcomponent name. Returns: Forward the result from the wrapped function. Raises: ValueError: if component and subcomponent name is not valid. zGiven component name not foundz!Given subcomponent name not foundz+component and subcomponent name don't matchr5cs,j|_|js j|_t}||dSr)rr subcomponentrrTFLiteConverterMetricsset_converter_error)r5tflite_metrics)rr@r r report_errors z#convert_phase..report_errorr;cst}||_|dSr)rr8r;)r;r5)rDr rreport_error_messages z+convert_phase..report_error_messagecstfdd}|S)Nc sz|i|WSty-}z|jr |jD]}|q|dt||dd}~wtyA}z t||dd}~wwr)r-r0str Exception)argskwargsconverter_errorr5error)funcrDrEr rwrappers    z8convert_phase..actual_decorator..wrapper) functoolswraps)rLrM)rDrE)rLractual_decoratorsz'convert_phase..actual_decorator)r ValueErrorrrrrr8r)rr@rPr )rrDrEr@r convert_phases    rR)r collectionsenumrNtypingr%tensorflow.compiler.mlir.lite.metricsrtensorflow.lite.python.metricsrEnumr namedtuplerrrGr-rrRr r r rs    T'