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 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
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 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