Searched defs:TorchScriptTensor (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/onnx/_internal/fx/ |
| D | fx_onnx_interpreter.py | 90 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
224 onnxscript_values: onnxscript_graph_building.TorchScriptTensor
337 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
402 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
558 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
611 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
676 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
704 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
777 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
|
| D | diagnostics.py | 183 def _onnxscript_torch_script_tensor(obj: graph_building.TorchScriptTensor) -> str:
|
| D | onnxfunction_dispatcher.py | 664 attribute: fx_type_utils.Argument | onnxscript_graph_building.TorchScriptTensor,
|