onnxruntime/onnxruntime/python
Vincent Wang f0f3012666
Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input (#7899)
* add SoftmaxCrossEntropyLossInternal

* bugfix and ut

* fix ut

* fix ut

* support torch1.8.1

* function body for nll_loss_internal
2021-06-09 10:29:46 +08:00
..
backend fix py packaging pipeline (#5038) 2020-09-03 09:32:10 -07:00
datasets replace invalid sample (#4567) 2020-07-21 23:51:17 -07:00
tools Add SoftmaxCrossEntropyLossInternal to Support Dynamic ignore_index Input (#7899) 2021-06-09 10:29:46 +08:00
training Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_ld_preload.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
_pybind_state.py Fix path used in check for cudnn library (#7786) 2021-05-28 09:32:13 +10:00
exported_symbols.lst
onnxruntime_collect_build_info.py Liqun/ort training version (#7620) 2021-05-14 09:54:19 -07:00
onnxruntime_inference_collection.py Don't hold onto unnecessary numpy references while binding numpy objectas as inputs (#7881) 2021-05-30 21:12:32 -07:00
onnxruntime_pybind_exceptions.cc Python bindings fix ups in preparation to Sparse Tensor introduction (#7817) 2021-05-26 09:47:41 -07:00
onnxruntime_pybind_exceptions.h Python bindings fix ups in preparation to Sparse Tensor introduction (#7817) 2021-05-26 09:47:41 -07:00
onnxruntime_pybind_iobinding.cc Python bindings fix ups in preparation to Sparse Tensor introduction (#7817) 2021-05-26 09:47:41 -07:00
onnxruntime_pybind_mlvalue.cc Python bindings fix ups in preparation to Sparse Tensor introduction (#7817) 2021-05-26 09:47:41 -07:00
onnxruntime_pybind_mlvalue.h Python bindings fix ups in preparation to Sparse Tensor introduction (#7817) 2021-05-26 09:47:41 -07:00
onnxruntime_pybind_ortvalue.cc training with custom autograd Functions (#7513) 2021-06-07 13:01:21 -07:00
onnxruntime_pybind_state.cc Python with DmlExecutionProvider : choose device_id in SessionOptions (#7964) 2021-06-08 13:02:12 -07:00
onnxruntime_pybind_state_common.cc training with custom autograd Functions (#7513) 2021-06-07 13:01:21 -07:00
onnxruntime_pybind_state_common.h training with custom autograd Functions (#7513) 2021-06-07 13:01:21 -07:00
onnxruntime_validation.py Liqun/ort training version (#7620) 2021-05-14 09:54:19 -07:00
pybind.def
version_script.lds