onnxruntime/onnxruntime/python
PeixuanZuo 326cf2f5e9
[ROCm] add Softmax Tunable Op (#14541)
### Description
Add Softmax Tunable Op, only include blockwise vec implementation and
composable kernel.
Related PR: https://github.com/microsoft/onnxruntime/pull/14475,
https://github.com/microsoft/onnxruntime/pull/14612

---------

Co-authored-by: peixuanzuo <peixuanzuo@linmif39a000004.zvflicr54joexhdgnhvmxrxygg.phxx.internal.cloudapp.net>
2023-02-13 15:56:50 +08:00
..
backend replace 'master' branch ref to 'main' for onnx repo (#12678) 2022-08-30 13:41:42 -07:00
datasets
providers/tvm Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
tools [ROCm] add Softmax Tunable Op (#14541) 2023-02-13 15:56:50 +08:00
torch_cpp_extensions [ORTModule] ATen Support for aten::upsample_nearest (#13364) 2022-10-20 08:30:04 +08:00
training
__init__.py
_ld_preload.py
_pybind_state.py.in Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460) 2022-08-22 09:40:40 -07:00
exported_symbols.lst
numpy_helper.h Fix SDL warnings in CPU EP (#9975) 2021-12-19 20:54:29 -08:00
onnxruntime_collect_build_info.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnxruntime_inference_collection.py Enable ORT in TorchDynamo (#13259) 2022-11-01 11:19:29 -07:00
onnxruntime_pybind.h fix windows ci debug build break (#11495) 2022-05-12 16:54:00 -07:00
onnxruntime_pybind_exceptions.cc
onnxruntime_pybind_exceptions.h Fix issue with debug VS2022 build when python bindings are enabled (#9794) 2021-11-18 16:58:02 +10:00
onnxruntime_pybind_iobinding.cc Adds missing numpy type when looking for the ort correspondance (#10943) 2022-03-22 14:44:48 -07:00
onnxruntime_pybind_mlvalue.cc Multi-stream execution support (#13495) 2022-12-15 07:39:29 -08:00
onnxruntime_pybind_mlvalue.h Move OrtValueVector from onnxruntime-training to onnxruntime (#11176) 2022-06-15 09:36:28 +02:00
onnxruntime_pybind_module.cc Fix issue with debug VS2022 build when python bindings are enabled (#9794) 2021-11-18 16:58:02 +10:00
onnxruntime_pybind_ortvalue.cc Enable ORT in TorchDynamo (#13259) 2022-11-01 11:19:29 -07:00
onnxruntime_pybind_schema.cc Pass SessionOptions to XnnpackProviderFactoryCreator. (#13318) 2022-12-10 14:23:46 +08:00
onnxruntime_pybind_sparse_tensor.cc Multi-stream execution support (#13495) 2022-12-15 07:39:29 -08:00
onnxruntime_pybind_state.cc Adding RunOptions synchronization behaviour to C/C++ API (#14088) 2023-02-07 19:59:28 -08:00
onnxruntime_pybind_state.h Fix issue with debug VS2022 build when python bindings are enabled (#9794) 2021-11-18 16:58:02 +10:00
onnxruntime_pybind_state_common.cc Allow CUDA EP enable or disable TunableOp via session options and environment variable (#13601) 2022-11-15 14:43:54 +08:00
onnxruntime_pybind_state_common.h [oneDNN] Improved thread handling (#13618) 2023-01-31 14:37:13 -08:00
onnxruntime_validation.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
pybind.def
version_script.lds
version_script_expose_onnx_protobuf.lds