AVh3dZddlmZddlmZddlmZddlmZddlm Z ddl m Z e dg d Z d Z d Zd ZdZy)z3Decorator to overrides the gradient for a function.)pywrap_tf_session)cpp_shape_inference_pb2)dtypes)core)compat) tf_exportz)__internal__.ops.get_resource_handle_data)v1ct|tjrt|tjrJ|jj j 5}tj||j}dddtjjjtjS#1swYExYw)N) isinstancerSymbolValuegraph_c_graphgetrGetHandleShapeAndType _as_tf_outputrCppShapeInferenceResult HandleData FromStringras_bytes)graph_opc_graph handle_datas V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/handle_data_util.pyget_resource_handle_datars Xt{{ +Xtzz24 3~~""$+#99'')+K+ ! 8 8 C C N N ook" $$ ++s %CC cdt|tjr |jSt |S)z!Obtains HandleData from a tensor.)r rr _handle_datar)source_ts rget_handle_datar&s'$**%   !( ++c|jtjk(s|jtjk(rt |}t ||yy)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_trs rcopy_handle_datar'-s=nn'nn&!(+KHk*'r cV||jr |jsyt|tjr||_y|j jj5}tj||j|jdddy#1swYyxYw)z%Sets handle data on the giver tensor.N) is_setshape_and_typer rr rrrrrSetHandleShapeAndTyperSerializeToString)r&rrs rr%r%Bs     ' ' $**%'H ~~""$M++GX5K5K5M,7,I,I,KMMMMs "4BB(ctjj}d|_|jj tjj |j|j|S)NT)shaper") rrrr)r*appendHandleShapeAndTypeas_protoas_datatype_enum)r.r"rs rcreate_handle_datar3Ush'??JJL++ ##55HH (>(>I@A r N)__doc__tensorflow.python.clientrtensorflow.python.frameworkrrtensorflow.python.typesrtensorflow.python.utilr tensorflow.python.util.tf_exportrrrr'r%r3r rr;sM:6?.()6 62> $? $,+*M&r