Searched defs:dynamic_axes (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | export.cpp | 509 std::unordered_map<int64_t, std::string>>& dynamic_axes, in GraphEncoder()
611 std::unordered_map<int64_t, std::string>>& dynamic_axes, in TensorTypeToONNXType()
649 std::unordered_map<int64_t, std::string>>& dynamic_axes) { in EncodeValueInfoType()
728 std::unordered_map<int64_t, std::string>>& dynamic_axes) { in EncodeValueInfo()
740 std::unordered_map<int64_t, std::string>>& dynamic_axes, in EncodeGraph()
762 std::unordered_map<int64_t, std::string>>& dynamic_axes, in EncodeBlock()
1407 std::unordered_map<std::int64_t, std::string>>& dynamic_axes, in export_onnx()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | utils.py | 566 dynamic_axes=None, argument
1086 dynamic_axes=None, argument
1225 dynamic_axes=None, argument
1473 dynamic_axes=None, argument
1943 def _validate_dynamic_axes(dynamic_axes, model, input_names, output_names): argument
|
| H A D | _experimental.py | 24 dynamic_axes: Optional[Mapping[str, Union[Mapping[int, str], Sequence[int]]]] = None variable in ExportOptions
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _compat.py | 29 def _from_dynamic_axes_to_dynamic_shapes( argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_onnx_opset.py | 58 dynamic_axes=None, argument
|
| H A D | verify.py | 243 dynamic_axes=None, argument
|
| H A D | onnx_test_common.py | 154 dynamic_axes=None, argument
|
| H A D | test_utility_funs.py | 68 dynamic_axes=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.cpp | 260 const NodeAttrNameMap& node_attr_to_name) { in initPythonIRBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 2220 std::unordered_map<int64_t, std::string>>& dynamic_axes, in ONNXSetDynamicInputShape()
|