BVh!~dZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZdZeddgej4dedZdZdZdZeddgej4deddZddZ dZ!y) 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)v1NcVtjr tdt|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)tensors X/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/utils_impl.pybuild_tensor_infor)s-0  $ %% #F ++ct|tjr?t|tjs%t|t j s t|Stjtj|jj|jj}t|tjrq|j j"|j$_|j(j"|j$_|j,j"|j$_|S|j"|_|S)z9Utility function to build TensorInfo proto from a Tensor.)dtyper ) isinstancerCompositeTensorr SparseTensorr ResourceVariable%_build_composite_tensor_info_internalr TensorInfor as_dtyperas_datatype_enum get_shapeas_protovaluesname coo_sparsevalues_tensor_nameindicesindices_tensor_name dense_shapedense_shape_tensor_name)r tensor_infos rrrGs)99: V]77 8 V2CC D 0 88)) OOFLL ) : :##%..02+ 22306 0B0BK-171D1DK.5;5G5G5L5LK2 {{K rc~|j}tj}tj|}|j j j|jtj|dD]>}|j jjjt|@|S)zBUtility function to build TensorInfo proto from a CompositeTensor.Texpand_composites) _type_specrr"rencode_structurer type_specCopyFromtype_spec_valuerflatten componentsaddr)rspecr/ spec_proto components rr!r!Zs   $))++%66t<* ((11*2L2LM<<$?/i  ++//1::"9-// rctjr tdtjt j tjj|jS)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() rrrrr"r DT_INVALIDr unknown_shaper&r()ops rbuild_tensor_info_from_oprBfs[@  J LL  " "  --/88: 77 rz'saved_model.get_tensor_from_tensor_infoz-saved_model.utils.get_tensor_from_tensor_infocxstjfd}|jd}|dk(r||jS|dk(ret j ||j j||j j||j jS|dk(rtj|jj}tj|}|jj Dcgc]}||j}}t#j$||dSt'd |d cc}w) 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. cPjtj|S)N import_scope)get_tensor_by_namer prepend_name_scope)r(graphrFs r _get_tensorz0get_tensor_from_tensor_info.._get_tensors(  # # t,? AArencodingr(r)r)r7Tr1zInvalid TensorInfo.encoding: zJ. Expected `coo_sparse`, `composite_tensor`, or `name` for a dense tensor.)r get_default_graph WhichOneofr(r rr)r,r*r.rStructuredValuerr5r decode_protor9rpack_sequence_as ValueError) r/rIrFrJrKr<r;r=r9s `` rget_tensor_from_tensor_inforRsC,  *3((*%A # #J /(  {'' ((<  % %K**>>?K**==>K**BBC EE%%++#44>>@J ! . .z :D..99;)+inn-;J;  zT JJ 4XJ?  ;sEc|xstj}|jtj|j|S)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` rE)r rLas_graph_elementrHr()r/rIrFs rget_element_from_tensor_inforUsB  *3((*%    [--LI KKrc2tj|||y)N)bst%swap_tensor_content_in_graph_function)meta_graph_def from_endiness to_endinesss rswap_function_tensor_contentr\s++m[r)NN)"__doc__tensorflow.core.frameworkrtensorflow.core.protobufrrtensorflow.python.eagerrtensorflow.python.frameworkrrWrr r r r tensorflow.python.opsr tensorflow.python.saved_modelrtensorflow.python.utilrr tensorflow.python.util.tf_exportr_DEPRECATION_MSG deprecatedrrr!rBrRrUr\rrris3/3/+?8.+547@.'6 *  ')NOQ./,0Q,6& &R 8>@A./) 0A) XK*r