U Ic_@sLdZddlmZmZddlZddlmZddlmZeeedddZdS) z Tools to help with tensor property propagation. This is not intended to be imported directly; please use the exposed functionalities in `torch.jit`. )AnyListN) TensorType)Graph)graph example_inputcCst|}t|dkrdS|d}t|tjjrN|dkrN|dd}t|t|ksft dt t ||D]`\}\}}|dkrqtt|tj t|t krt d|d||t|tj rt|t |qtdS)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 isinstancetypetorchZ_CZ ClassTypeZ debugName RuntimeError enumeratezipZTensorrZsetTypeZcreate_from_tensor)rrZ graph_inputsZin_0iZgraph_iZ example_irK/tmp/pip-unpacked-wheel-gikjz4vx/torch/jit/_passes/_property_propagation.pyapply_input_props_using_examples"    r) __doc__typingrrrrZtorch._Crrrrrrs