Home
last modified time | relevance | path

Searched refs:onnxscript_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Ddecomposition_skip.py37 onnxscript_function: Callable
89 onnxscript_function = torchlib_nn.aten_upsample_bilinear2d_vec # type: ignore[attr-defined] variable in UpsampleBilinear2DDecompSkip
104 function=cls.onnxscript_function,
127 onnxscript_function = torchlib_nn.aten_upsample_trilinear3d_vec # type: ignore[attr-defined] variable in UpsampleTrilinear3DDecompSkip
142 function=cls.onnxscript_function,
165 onnxscript_function = torchlib_core.aten_instance_norm # type: ignore[attr-defined] variable in InstanceNormDecompSkip
186 function=cls.onnxscript_function,
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_core.py500 for identifier, onnxscript_function in tracer.functions.items():
504 proto = onnxscript_function.to_function_proto()