BVhxdZddlmZddlmZdZdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ d Z!d!Z"d"Z#d#Z$d$Z%d%Z&d&Z'd'Z(d(Z)e*eeee e eeeeee%eh Z+e*jYe+ee&hZ-e*jYe-ee e e eeeeeeeeeeee e!e"e#e$e'e(e)hZ.d)Z/d,d+Z0y*)-zDThis file lists FunctionDef attributes and corresponding allowlists.)attr_value_pb2)compatapi_implementsapi_preferred_devicebackward_function_name _disable_acd_disable_call_shape_inferencedisable_summaries_at_runtime_construction_contextforward_function_name go_backwards _implements _input_shapesexperimental_ints_on_device _noinline_original_func_name_OutputsOnOpDeviceztf_quant.composite_functionztf_quant.quantized_opsruntime_constant_optimizationshared_rendezvous_tf_data_function_tftrt_allow_build_at_runtime_tftrt_convert_function_tftrt_is_dyn_op_tftrt_trt_logger_name_tftrt_max_batch_size_tftrt_max_cached_engines_tftrt_max_workspace_size_bytes_tftrt_minimum_segment_size_tftrt_precision_mode_tftrt_profile_strategy_tftrt_use_calibration_tftrt_use_implicit_batch time_major_XlaMustCompile _XlaCompile _XlaScope_XlaSeparateCompiledGradientsc 0t|tjr|St|trtj|St|trtj|St|t rtj|St|t tfr)tjtj|St|trtjj}|D]}t|tr|jj|/t|tr|jj|[t|t r|jj|t|t tfr/|j jtj|t#d|dt%|dtj|St#d |d t%|d) z?Converts a python object to an attr_value_pb2.AttrValue object.)b)i)f)szAttributes for z* must be bool, int, float, or string. Got .)listz Attribute z; must be bool, int, float, string, list, or AttrValue. Got ) isinstancer AttrValueboolintfloatstrbytesras_bytesr. ListValuer)appendr*r+r, ValueErrortype)keyvalue list_valuevs g/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/polymorphic_function/attributes.py_parse_func_attr_valuer@ys~//0 L%  # #e ,,%  # #e ,,%  # #e ,,%#u&  # #fooe&< ==%))335J    At  A a  A a  A a#u & FOOA./cU#7)1       # # 44  SEu+a ) Nc|st}i}|jD]'\}}||vrtd|dt||||<)|S)aConvert the keyword arguments into function_def attributes. Currently only support primitive types: bool, int, float and string. Args: attributes: the dictionary of attributes. allowlist: set of attribute names allowed. Returns: A dict of attributes where the key is the name of attribute and the value is the AttrValue proto. Raises: ValueError: If the kwargs contains unallowlisted name or unsupported value types. zAllowlist does not support `z` as an attribute.)MONOMORPHIC_FUNCTION_ALLOWLISTitemsr9r@) attributes allowlistattrsr;r<s r?parse_func_attrsrHsh .I %$$&4jc5 )  (-? @ BB'U3E#J 4 ,rA)N)1__doc__tensorflow.core.frameworkrtensorflow.python.utilrAPI_IMPLEMENTSAPI_PREFERRED_DEVICEBACKWARD_FUNCTION DISABLE_ACDDISABLE_CALL_SHAPE_INFERENCEDISABLE_SUMMARIES_AT_RUNTIME"EAGER_RUNTIME_CONSTRUCTION_CONTEXTFORWARD_FUNCTION GO_BACKWARDS IMPLEMENTS INPUT_SHAPESINTS_ON_DEVICE NO_INLINEORIGINAL_FUNCTION_NAMEOUTPUTS_ON_OP_DEVICEQUANTIZED_COMPOSITE_FUNCTION QUANTIZED_OPSRUNTIME_CONSTANT_OPTIMIZATIONSHARED_RENDEZVOUSTF_DATA_FUNCTIONTFTRT_ALLOW_BUILD_AT_RUNTIMETFTRT_CONVERT_FUNCTIONTFTRT_IS_DYN_OP TFTRT_LOGGERTFTRT_MAX_BATCH_SIZETFTRT_MAX_CACHED_ENGINESTFTRT_MAX_WORKSPACE_SIZETFTRT_MIN_SEGMENT_SIZETFTRT_PRECISION_MODETFTRT_PROFILE_STRATEGYTFTRT_USE_CALIBRATIONTFTRT_USE_IMPLICIT_BATCH TIME_MAJOR XLA_COMPILEXLA_COMPILE_OPTIONAL XLA_SCOPEXLA_SEPERATE_COMPILED_GRADIENTS frozensetPOLYMORPHIC_FUNCTION_ALLOWLISTunionTRACING_COMPILATION_ALLOWLISTrCr@rHrAr?rvs K4)"-, >=%<"*   .  .+<( ?'&>2$' .6<6.206   $  "A!*  ! , "!* 1 1"!"+!2!2!$*$$   '/"<"JrA