Home
last modified time | relevance | path

Searched defs:TorchScriptTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dfx_onnx_interpreter.py90 | 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, ...],
H A Ddiagnostics.py183 def _onnxscript_torch_script_tensor(obj: graph_building.TorchScriptTensor) -> str:
H A Donnxfunction_dispatcher.py664 attribute: fx_type_utils.Argument | onnxscript_graph_building.TorchScriptTensor,