o ?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_graphgetrZGetHandleShapeAndType _as_tf_outputrCppShapeInferenceResult HandleDataZ FromStringras_bytes)Zgraph_opc_graph handle_datarg/home/www/facesmatcher.com/pyenv/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  rcCsv|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 rZSetHandleShapeAndTyper ZSerializeToString)rrrrrrr@s  "rcCs2tj}d|_|jtjj||jd|S)NT)shaper) rrrrr appendZHandleShapeAndTypeZas_protoZas_datatype_enum)r!rrrrrcreate_handle_dataSs  r#N)__doc__Ztensorflow.python.clientrZtensorflow.python.frameworkrrZtensorflow.python.typesrZtensorflow.python.utilrrrrrr#rrrrs