BVhUdZddlZddlZddlZddlmZddlmZddlm Z GddejZ ejdd d gZ Gd d e ejZGd deZej"fdZy)z(Utilities for collecting TFLite metrics.N)Text)converter_error_data_pb2)metricsceZdZdZdZdZdZy) Componentz5Enum class defining name of the converter components.PREPARE_TF_MODELCONVERT_TF_TO_TFLITE_MODELOPTIMIZE_TFLITE_MODELN)__name__ __module__ __qualname____doc__rr r T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/lite/python/convert_phase.pyrrs=' <2rrSubComponentItemname componentceZdZdZdZedZedZeddZ ede jZ ede jZ ed e jZed e jZed e jZed e jZed e jZede jZede j(Zede j(Zede j(Zede j(Zede j2Zede j2Zede j2Zede j2Zy) SubComponentzEnum class defining name of the converter subcomponents. This enum only defines the subcomponents in Python, there might be more subcomponents defined in C++. c.|jjSNvaluerselfs r__str__zSubComponent.__str__1s ::??rc.|jjSrrrs rrzSubComponent.name4s ::??rc.|jjSr)rrrs rrzSubComponent.component8s ::  r 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 rrpropertyrrrr rrr!r"r#r$r%r&r'r(r r)r*r+r,r r-r.r/r0rrrrr*s       !5+%%6%.%?%?A/&&8&/&@&@B((<(1(B(BD"2$i&@&@"B/?193M3M/O+((<(1(B(BD..H.7.H.HJ'':'0'A'AC1A3**1,- &&8&/&J&JL))>)2)M)MO%%6%.%I%IK/)9+Y-L-L)N%{I,K,KL)j)*I*I J(j)*I*I J(rrcHeZdZdZfdZdej fdZdZxZ S)ConverterErrorz4Raised when an error occurs during model conversion.c\tt| |g|_|j |yr)superr3__init__errors_parse_error_message)rmessage __class__s rr6zConverterError.__init__s' .$(1DKg&r error_datac:|jj|yr)r7append)rr;s r append_errorzConverterError.append_errorsKKz"rcdtjji}|jD]>\}}||vs tj}||_||_|j |yy)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_coder>)rr9error_code_mappingpatternrDr;s rr8z#ConverterError._parse_error_messagesv / % 7 7 . .  2779 G -@@B #*  *  *% r) r r r rr6rr@r>r8 __classcell__)r:s@rr3r3~s#<' #7JJ#rr3ctvr tdtvr tdtjk7rjk7r tddt j ffd dtffd fd}|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 matchr;cj|_|jsj|_t j }|j |yr)rr subcomponentrrTFLiteConverterMetricsset_converter_error)r;tflite_metricsrrJs r report_errorz#convert_phase..report_errorsD%??J  " " , 1 1j335N&&z2rrCcLtj}||_|yr)rr@rC)rCr;rNs rreport_error_messagez+convert_phase..report_error_messages!)<<>J,JrcHtjfd}|S)Nc |i|S#t$rA}|jr|jD] }| |dt||dd}~wt$r}t||dd}~wwxYwr)r3r7str Exception)argskwargsconverter_errorr;errorfuncrNrPs rwrapperz8convert_phase..actual_decorator..wrappers T$V$$ (  ! !+22 %j  $ %4' s?3 44' SZ(s  A5.actual_decorators%__T   Nr)r ValueErrorrr rrr@r)rrJr]rNrPs`` @@r convert_phaser_si 5 66% 8 99l... ) B CC37JJ3$ & r)r collectionsenumr[typingr%tensorflow.compiler.mlir.lite.metricsrtensorflow.lite.python.metricsrEnumr namedtuplerrrTr3r r_rrrrgs/ J2 2 2*;))*<+1;*?AQK#TYYQKh$Y$N+7*B*B6r