Home
last modified time | relevance | path

Searched refs:export_to_pretty_string (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_no_runtime.py118 torch.onnx.export_to_pretty_string(traced, (torch.rand([2]),))
128 torch.onnx.export_to_pretty_string(mte, (torch.zeros(1, 2, 3),), verbose=False)
141 torch.onnx.export_to_pretty_string(
182 torch.onnx.export_to_pretty_string(mte, (torch.zeros(1, 2, 3),), verbose=False)
201 torch.onnx.export_to_pretty_string(mte, (torch.zeros(1, 2, 3),), verbose=False)
215 torch.onnx.export_to_pretty_string(mte, (torch.zeros(1, 2, 3),), verbose=False)
227 torch.onnx.export_to_pretty_string(
239 torch.onnx.export_to_pretty_string(mte, (torch.rand(3, 4),), verbose=False)
250 torch.onnx.export_to_pretty_string(mte, (torch.zeros(1, 2, 3),), verbose=False)
280 torch.onnx.export_to_pretty_string(mte, (torch.ones(2, 3),), verbose=False)
[all …]
H A Dtest_operators.py53 return torch.onnx.export_to_pretty_string(
/aosp_15_r20/external/pytorch/torch/onnx/
H A D__init__.py75 export_to_pretty_string,
H A Dutils.py1209 def export_to_pretty_string( function
/aosp_15_r20/external/pytorch/docs/source/
H A Donnx_torchscript.rst700 .. autofunction:: export_to_pretty_string