Home
last modified time | relevance | path

Searched defs:onnx_shape_inference (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_custom_ops.py15 onnx_shape_inference = True variable in TestCustomAutogradFunction
88 onnx_shape_inference = True variable in TestExportAsContribOps
H A Dtest_autograd_funs.py16 onnx_shape_inference = True variable in TestAutogradFuns
/aosp_15_r20/external/pytorch/torch/csrc/onnx/
H A Dinit.cpp51 int opset_version) { in initONNXBindings()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp2282 bool onnx_shape_inference) { in ONNXUpdateTypeFromTensor()
2299 bool onnx_shape_inference, in ONNXAssignOutputShape()
2463 bool onnx_shape_inference, in ONNXAssignOutputShape()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dutils.py1478 onnx_shape_inference=True, argument