Home
last modified time | relevance | path

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

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