VhNdZddlmZddlZddlmZddlmZdedeeddfd Zy) z Tools to help with tensor property propagation. This is not intended to be imported directly; please use the exposed functionalities in `torch.jit`. )AnyN) TensorType)Graphgraph example_inputreturnct|j}t|dk(ry|d}t|j t j jr|jdk(r|dd}t|t|k(s tdtt||D]\}\}}| t|t jt|j tk7rtd|d||t|t jsq|jtj|y)z` Applies properties for each tensor in the graph inputs using the example supplied. rNselfzHNumber of inputs in graph does not match number of inputs in the examplezInput z does not match type of example)listinputslen isinstancetypetorch_C ClassType debugName RuntimeError enumeratezipTensorrsetTypecreate_from_tensor)rr graph_inputsin_0igraph_i example_is W/home/dcms/DCMS/lib/python3.12/site-packages/torch/jit/_passes/_property_propagation.pyapply_input_props_using_exampler!s#  'L <A ?D$))+uxx112t~~7G67Q#AB' | M 2 2 V  $-S}-M#N F GY    i .* LLNJ3  :;Wi  i . OOJ99)D E F) __doc__typingrrrtorch._Crr r!r"r r's: F5FcFtFr"