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 Add a couple of utility scripts to tools/python (#3621) 2020-05-02 17:35:43 +10:00
convert_onnx_models_to_ort.py Android package infrastructure (#7430) 2021-04-30 14:23:54 +10:00
create_reduced_build_config.py Support specifying globally allowed types from build script (#6677) 2021-02-22 14:05:00 -08:00
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 The fixings for python scripts in ONNXRuntime (#4135) 2020-06-08 10:27:32 -07:00
example_operator_perf_test.py Downgrade GCC (#5269) 2020-09-24 21:14:54 -07:00
FindOptimizerOpsetVersionUpdatesRequired.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
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 Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
onnx_test_data_utils.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
ort_test_dir_utils.py Add detailed assertion error message (#7232) 2021-04-05 10:05:40 -07:00
PythonTools.md Add a few more things to the helper python scripts. (#3842) 2020-05-12 09:59:40 +10:00
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 Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
run_android_emulator.py Add support for running Android emulator from build.py on Windows. (#6317) 2021-01-13 19:21:49 -08:00
sparsify_initializers.py Fix bugs in sparsify script (#7134) 2021-03-25 14:53:52 -07:00
update_version.py allow update_version.py to update new npm packages (#7746) 2021-05-18 16:15:19 -07:00