Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dfx_onnx_interpreter.py89 onnxscript_graph_building.TorchScriptTensor
90 | tuple[onnxscript_graph_building.TorchScriptTensor, ...], argument
118 onnxscript_graph_building.TorchScriptTensor
119 | tuple[onnxscript_graph_building.TorchScriptTensor, ...]
140 element_value, onnxscript_graph_building.TorchScriptTensor
176 onnxscript_graph_building.TorchScriptTensor
178 | tuple[onnxscript_graph_building.TorchScriptTensor, ...]
224 onnxscript_values: onnxscript_graph_building.TorchScriptTensor argument
225 | tuple[onnxscript_graph_building.TorchScriptTensor, ...],
336 onnxscript_graph_building.TorchScriptTensor
[all …]
H A Ddiagnostics.py183 def _onnxscript_torch_script_tensor(obj: graph_building.TorchScriptTensor) -> str: argument
H A Donnxfunction_dispatcher.py664 attribute: fx_type_utils.Argument | onnxscript_graph_building.TorchScriptTensor, argument