o i e @sldZddlmZddlmZddlmZddlmZddlm Z ddZ d d Z d d Z d dZ ddZdS)z3Decorator to overrides the gradient for a function.)pywrap_tf_session)cpp_shape_inference_pb2)dtypes)core)compatcCslt|tjr t|tjrJ|jj}t|| }Wdn1s'wYt j j t|S)N) isinstancerSymbolValuegraph_c_graphgetrGetHandleShapeAndType _as_tf_outputrCppShapeInferenceResult HandleData FromStringras_bytes)graph_opc_graph handle_datarU/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/ops/handle_data_util.pyget_resource_handle_datas  rcCst|tjr |jSt|S)z!Obtains HandleData from a tensor.)rrr _handle_datar)source_trrrget_handle_data$s rcCs2|jtjks |jtjkrt|}t||dSdS)aCopies HandleData for variant and resource type tensors if available. The CppShapeInferenceResult::HandleData proto contains information about the shapes and types of the element tensors of resource/variant type tensors. We need to copy this across function boundaries, i.e., when capturing a placeholder or when returning a function tensor as output. If we don't do this the element tensors will have unknown shapes, e.g., if a TensorList variant tensor is captured as a placeholder, elements popped from that list would have unknown shape. Args: source_t: The tensor to copy HandleData from. target_t: The tensor to copy HandleData to. N)dtyperresourcevariantrset_handle_data)rtarget_trrrrcopy_handle_data+s  r!cCsv|dus |jr |js dSt|tjr||_dS|jj}t || | WddS1s4wYdS)z%Sets handle data on the giver tensor.N) is_setshape_and_typerrr rr r r rSetHandleShapeAndTyperSerializeToString)r rrrrrr@s  "rcCs2tj}d|_|jtjj||jd|S)NT)shaper) rrrr"r#appendHandleShapeAndTypeas_protoas_datatype_enum)r&rrrrrcreate_handle_dataSs  r+N)__doc__tensorflow.python.clientrtensorflow.python.frameworkrrtensorflow.python.typesrtensorflow.python.utilrrrr!rr+rrrrs