o ?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)tensorri/home/www/facesmatcher.com/pyenv/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 ) isinstancerZCompositeTensorr SparseTensorr ZResourceVariable%_build_composite_tensor_info_internalr TensorInforZas_dtyperZas_datatype_enum get_shapeas_protovaluesname coo_sparsevalues_tensor_nameindicesindices_tensor_nameZ 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.TZexpand_composites)Z _type_specrrr Zencode_structurer type_specZCopyFromtype_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!) rrrrrrZ DT_INVALIDr Z unknown_shaperr!)oprrrbuild_tensor_info_from_opfs  r2z'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)Zget_tensor_by_namer prepend_name_scoper!)graphr4rr _get_tensors z0get_tensor_from_tensor_info.._get_tensorencodingr!r"r)r*csg|]}|jqSrr6).0r0)r8rr 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_graphZ WhichOneofr!r rr"r%r#r&rZStructuredValuerr)r Z decode_protor,rZpack_sequence_as ValueError)r'r7r4r9r/r.r,r)r8r7r4rget_tensor_from_tensor_infos*        r>cCs"|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` r3)r r<Zas_graph_elementr5r!)r'r7r4rrrget_element_from_tensor_infos r?cCst|||dS)N)bstZ%swap_tensor_content_in_graph_function)Zmeta_graph_defZ from_endinessZ to_endinessrrrswap_function_tensor_contentsrA)NN)"__doc__Ztensorflow.core.frameworkrZtensorflow.core.protobufrrZtensorflow.python.eagerrZtensorflow.python.frameworkrr@rrr r r Ztensorflow.python.opsr Ztensorflow.python.saved_modelr Ztensorflow.python.utilrrZ tensorflow.python.util.tf_exportrZ_DEPRECATION_MSG deprecatedrrrr2r>r?rArrrrsB                  )  ,