You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@mxnet.apache.org by ke...@apache.org on 2019/10/20 23:23:58 UTC

[incubator-mxnet] branch master updated (217ae02 -> 746cbc5)

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 217ae02  Fix numpy bugs (#16537)
     add 746cbc5  Add unit tests for TensorRT integration and fix some bugs (#15399)

No new revisions were added by this update.

Summary of changes:
 3rdparty/onnx-tensorrt                             |   2 +-
 ci/docker/install/tensorrt.sh                      |   7 +-
 python/mxnet/contrib/tensorrt.py                   |   2 +
 src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h  |  15 +-
 src/operator/subgraph/tensorrt/nnvm_to_onnx.cc     |  43 +-
 src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc |   2 +-
 src/operator/subgraph/tensorrt/tensorrt-inl.h      | 119 ++++-
 src/operator/subgraph/tensorrt/tensorrt.cc         |   2 +-
 tests/python/tensorrt/common.py                    |  30 --
 tests/python/tensorrt/lenet5_common.py             |  31 --
 tests/python/tensorrt/lenet5_train.py              |  12 +-
 tests/python/tensorrt/test_cvnets.py               |   7 +-
 tests/python/tensorrt/test_ops.py                  | 527 +++++++++++++++++++--
 tests/python/tensorrt/test_resnet18.py             |   2 +-
 tests/python/tensorrt/test_tensorrt_batchnorm.py   |  65 ---
 tests/python/tensorrt/test_tensorrt_deconv.py      |  63 ---
 tests/python/tensorrt/test_tensorrt_lenet5.py      |  28 +-
 17 files changed, 687 insertions(+), 270 deletions(-)
 delete mode 100644 tests/python/tensorrt/common.py
 delete mode 100644 tests/python/tensorrt/lenet5_common.py
 delete mode 100644 tests/python/tensorrt/test_tensorrt_batchnorm.py
 delete mode 100644 tests/python/tensorrt/test_tensorrt_deconv.py