This is an automated email from the ASF dual-hosted git repository.
kellen pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 7bc3db8 Bump Scala version to 1.6 (#15660)
add 6b22aa4 handle fix_gamma in tensorrt subgraph conversion correctly
(#15645)
No new revisions were added by this update.
Summary of changes:
src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h | 21 ++++++--
src/operator/subgraph/tensorrt/nnvm_to_onnx.cc | 29 +++++++++-
src/operator/subgraph/tensorrt/tensorrt.cc | 2 +-
tests/python/tensorrt/test_tensorrt_batchnorm.py | 65 +++++++++++++++++++++++
4 files changed, 109 insertions(+), 8 deletions(-)
create mode 100644 tests/python/tensorrt/test_tensorrt_batchnorm.py
