onnxruntime/tools/python
Bowen Bao a776b57160
Add shape inference to custom symbolic functions (#7937)
**Description**: As title.

**Motivation and Context**
- PyTorch ONNX exporter heavily depends on ONNX shape inference to export accurate and efficient model. Custom symbolic function exports the op as contrib ops, thus exporter is unable to perform standard onnx shape inference. Models with dynamic shape inputs are affected.
2021-06-08 10:43:06 -07:00
..
util Add supported operators/types documentation for the ORT Mobile package (#7807) 2021-05-26 15:57:40 +10:00
__init__.py
convert_onnx_models_to_ort.py Android package infrastructure (#7430) 2021-04-30 14:23:54 +10:00
create_reduced_build_config.py
dump_ort_model.py Add script to dump initializer, NodeArg, Node and subgraph info from an ORT format model (#7516) 2021-05-04 08:34:35 +10:00
dump_subgraphs.py
example_operator_perf_test.py
FindOptimizerOpsetVersionUpdatesRequired.py
gen_contrib_doc.py Update the operator documentation generation (#7787) 2021-06-02 17:47:40 +10:00
gen_opkernel_doc.py Update the operator documentation generation (#7787) 2021-06-02 17:47:40 +10:00
gen_ort_mobile_pkg_doc.py Add supported operators/types documentation for the ORT Mobile package (#7807) 2021-05-26 15:57:40 +10:00
get_submodules.py
onnx_test_data_utils.py
ort_test_dir_utils.py Add detailed assertion error message (#7232) 2021-04-05 10:05:40 -07:00
PythonTools.md
register_custom_ops_pytorch_exporter.py Add shape inference to custom symbolic functions (#7937) 2021-06-08 10:43:06 -07:00
remove_initializer_from_input.py
run_android_emulator.py
sparsify_initializers.py
update_version.py allow update_version.py to update new npm packages (#7746) 2021-05-18 16:15:19 -07:00