Searched refs:onnx_shape_inference (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | utils.py | 633 if GLOBALS.onnx_shape_inference:
662 if GLOBALS.onnx_shape_inference:
1142 GLOBALS.onnx_shape_inference,
1164 GLOBALS.onnx_shape_inference,
1185 if GLOBALS.onnx_shape_inference:
1478 onnx_shape_inference=True, argument
1495 GLOBALS.onnx_shape_inference = onnx_shape_inference
|
| H A D | _globals.py | 33 self.onnx_shape_inference: bool = True
|
| H A D | verification.py | 675 if GLOBALS.onnx_shape_inference:
|
| H A D | symbolic_opset9.py | 6429 if GLOBALS.onnx_shape_inference:
6520 if GLOBALS.onnx_shape_inference:
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_custom_ops.py | 15 onnx_shape_inference = True variable in TestCustomAutogradFunction
88 onnx_shape_inference = True variable in TestExportAsContribOps
|
| H A D | test_models_onnxruntime.py | 44 self.onnx_shape_inference = True
87 onnx_shape_inference=True,
|
| H A D | test_autograd_funs.py | 16 onnx_shape_inference = True variable in TestAutogradFuns
|
| /aosp_15_r20/external/pytorch/torch/csrc/onnx/ |
| H A D | init.cpp | 49 bool onnx_shape_inference, in initONNXBindings()
56 onnx_shape_inference, in initONNXBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 2282 bool onnx_shape_inference) { in ONNXUpdateTypeFromTensor() argument
2283 if (onnx_shape_inference) { in ONNXUpdateTypeFromTensor()
2299 bool onnx_shape_inference, in ONNXAssignOutputShape() argument
2313 graph->outputs().at(outputs_index), var, onnx_shape_inference); in ONNXAssignOutputShape()
2322 onnx_shape_inference, in ONNXAssignOutputShape()
2371 onnx_shape_inference, in ONNXAssignOutputShape()
2388 onnx_shape_inference, in ONNXAssignOutputShape()
2463 bool onnx_shape_inference, in ONNXAssignOutputShape() argument
2474 onnx_shape_inference, in ONNXAssignOutputShape()
|
| H A D | shape_type_inference.h | 57 bool onnx_shape_inference,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
| H A D | jit_utils.py | 255 shape_inference=GLOBALS.onnx_shape_inference,
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 553 onnx_shape_inference: _bool,
|