o i e!@s0dZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZdZeddgdededdZddZddZddZeddgdeded&d d!Zd&d"d#Z d$d%Z!dS)'z,SavedModel utility functions implementation.) types_pb2)meta_graph_pb2) struct_pb2)context)byte_swap_tensor)composite_tensor)dtypes)ops) sparse_tensor) tensor_shape)resource_variable_ops)nested_structure_coder) deprecation)nest) tf_exportzThis API was designed for TensorFlow v1. See https://www.tensorflow.org/guide/migrate for instructions on how to migrate your code to TensorFlow v2.zsaved_model.build_tensor_infoz#saved_model.utils.build_tensor_info)v1NcCstrtdt|S)aUtility function to build TensorInfo proto from a Tensor. Args: tensor: Tensor or SparseTensor whose name, dtype and shape are used to build the TensorInfo. For SparseTensors, the names of the three constituent Tensors are used. Returns: A TensorInfo protocol buffer constructed based on the supplied argument. Raises: RuntimeError: If eager execution is enabled. @compatibility(TF2) This API is not compatible with eager execution as `tensor` needs to be a graph tensor, and there is no replacement for it in TensorFlow 2.x. To start writing programs using TensorFlow 2.x, please refer to the [Effective TensorFlow 2](https://www.tensorflow.org/guide/effective_tf2) guide. @end_compatibility z8`build_tensor_info` is not supported in eager execution.)rexecuting_eagerly RuntimeErrorbuild_tensor_info_internal)tensorrW/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/saved_model/utils_impl.pybuild_tensor_info)srcCst|tjrt|tjst|tjst|Stj t |j j |d}t|tjr@|jj|j_|jj|j_|jj|j_|S|j|_|S)z9Utility function to build TensorInfo proto from a Tensor.)dtyper ) isinstancerCompositeTensorr SparseTensorr ResourceVariable%_build_composite_tensor_info_internalr TensorInforas_dtyperas_datatype_enum get_shapeas_protovaluesname coo_sparsevalues_tensor_nameindicesindices_tensor_name dense_shapedense_shape_tensor_name)r tensor_inforrrrGs"         rcCsV|j}t}t|}|jj|jt j |ddD] }|jj t |q|S)zBUtility function to build TensorInfo proto from a CompositeTensor.Texpand_composites) _type_specrrr encode_structurer type_specCopyFromtype_spec_valuerflatten componentsaddr)rspecr, spec_proto componentrrrrZs  rcCs,trtdtjtjt |j dS)a:Utility function to build TensorInfo proto from an Op. Note that this function should be used with caution. It is strictly restricted to TensorFlow internal use-cases only. Please make sure you do need it before using it. This utility function overloads the TensorInfo proto by setting the name to the Op's name, dtype to DT_INVALID and tensor_shape as None. One typical usage is for the Op of the call site for the defunned function: ```python @function.defun def some_variable_initialization_fn(value_a, value_b): a = value_a b = value_b value_a = constant_op.constant(1, name="a") value_b = constant_op.constant(2, name="b") op_info = utils.build_op_info( some_variable_initialization_fn(value_a, value_b)) ``` Args: op: An Op whose name is used to build the TensorInfo. The name that points to the Op could be fetched at run time in the Loader session. Returns: A TensorInfo protocol buffer constructed based on the supplied argument. Raises: RuntimeError: If eager execution is enabled. z@`build_tensor_info_from_op` is not supported in eager execution.)rr r%) rrrrrr DT_INVALIDr unknown_shaper#r%)oprrrbuild_tensor_info_from_opfs  r=z'saved_model.get_tensor_from_tensor_infoz-saved_model.utils.get_tensor_from_tensor_infocsptfdd|d}|dkr|jS|dkr2t|jj|jj|jj S|dkrVt j |j j d}t|}fdd |j jD}tj||d d Std |d )ayReturns the Tensor or CompositeTensor described by a TensorInfo proto. Args: tensor_info: A TensorInfo proto describing a Tensor or SparseTensor or CompositeTensor. graph: The tf.Graph in which tensors are looked up. If None, the current default graph is used. import_scope: If not None, names in `tensor_info` are prefixed with this string before lookup. Returns: The Tensor or SparseTensor or CompositeTensor in `graph` described by `tensor_info`. Raises: KeyError: If `tensor_info` does not correspond to a tensor in `graph`. ValueError: If `tensor_info` is malformed. cstj|dS)N import_scope)get_tensor_by_namer prepend_name_scoper%)graphr?rr _get_tensors z0get_tensor_from_tensor_info.._get_tensorencodingr%r&r)r3csg|]}|jqSrrB).0r9)rDrr sz/get_tensor_from_tensor_info..Tr-zInvalid TensorInfo.encoding: zJ. Expected `coo_sparse`, `composite_tensor`, or `name` for a dense tensor.)r get_default_graph WhichOneofr%r rr&r)r'r+rStructuredValuerr1r decode_protor5rpack_sequence_as ValueError)r,rCr?rEr8r7r5r)rDrCr?rget_tensor_from_tensor_infos*        rNcCs"|pt}|tj|j|dS)aReturns the element in the graph described by a TensorInfo proto. Args: tensor_info: A TensorInfo proto describing an Op or Tensor by name. graph: The tf.Graph in which tensors are looked up. If None, the current default graph is used. import_scope: If not None, names in `tensor_info` are prefixed with this string before lookup. Returns: Op or tensor in `graph` described by `tensor_info`. Raises: KeyError: If `tensor_info` does not correspond to an op or tensor in `graph` r>)r rHas_graph_elementrAr%)r,rCr?rrrget_element_from_tensor_infos rPcCst|||dS)N)bst%swap_tensor_content_in_graph_function)meta_graph_def from_endiness to_endinessrrrswap_function_tensor_contentsrV)NN)"__doc__tensorflow.core.frameworkrtensorflow.core.protobufrrtensorflow.python.eagerrtensorflow.python.frameworkrrQrrr r r tensorflow.python.opsr tensorflow.python.saved_modelr tensorflow.python.utilrr tensorflow.python.util.tf_exportr_DEPRECATION_MSG deprecatedrrrr=rNrPrVrrrrsB                  )  ,