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 bde1b84 [int8] Add MobileNetV2_1.0 & ResNet18 Quantization (#14823)
add 1c874cf reformat trt to use subgraph API, add fp16 support (#14040)
No new revisions were added by this update.
Summary of changes:
3rdparty/onnx-tensorrt | 2 +-
ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt | 1 +
include/mxnet/c_api.h | 1 -
python/mxnet/contrib/tensorrt.py | 119 ++--
src/c_api/c_api_executor.cc | 86 +--
src/common/serialization.h | 318 -----------
src/executor/exec_pass.h | 12 -
src/executor/graph_executor.cc | 10 +
src/executor/graph_executor.h | 1 +
src/executor/tensorrt_pass.cc | 596 ---------------------
src/executor/trt_graph_executor.cc | 443 ---------------
src/executor/trt_graph_executor.h | 111 ----
src/operator/contrib/tensorrt-inl.h | 79 ---
src/operator/contrib/tensorrt.cc | 181 -------
src/operator/subgraph/build_subgraph.cc | 16 +-
.../tensorrt}/nnvm_to_onnx-inl.h | 131 ++---
.../{contrib => subgraph/tensorrt}/nnvm_to_onnx.cc | 281 ++++++----
.../subgraph/tensorrt}/onnx_to_tensorrt.cc | 27 +-
.../subgraph/tensorrt}/onnx_to_tensorrt.h | 18 +-
src/operator/subgraph/tensorrt/tensorrt-inl.h | 240 +++++++++
src/operator/subgraph/tensorrt/tensorrt.cc | 336 ++++++++++++
.../{contrib => subgraph/tensorrt}/tensorrt.cu | 30 +-
tests/cpp/misc/serialization.cc | 68 ---
tests/python/tensorrt/lenet5_train.py | 2 +
tests/python/tensorrt/test_cvnets.py | 34 +-
tests/python/tensorrt/test_cycle.py | 69 ---
tests/python/tensorrt/test_resnet18.py | 42 +-
tests/python/tensorrt/test_tensorrt_lenet5.py | 91 ++--
tests/python/tensorrt/test_training_warning.py | 70 ---
29 files changed, 1012 insertions(+), 2403 deletions(-)
delete mode 100644 src/common/serialization.h
delete mode 100644 src/executor/tensorrt_pass.cc
delete mode 100644 src/executor/trt_graph_executor.cc
delete mode 100644 src/executor/trt_graph_executor.h
delete mode 100644 src/operator/contrib/tensorrt-inl.h
delete mode 100644 src/operator/contrib/tensorrt.cc
rename src/operator/{contrib => subgraph/tensorrt}/nnvm_to_onnx-inl.h (61%)
rename src/operator/{contrib => subgraph/tensorrt}/nnvm_to_onnx.cc (68%)
rename src/{executor => operator/subgraph/tensorrt}/onnx_to_tensorrt.cc (89%)
rename src/{executor => operator/subgraph/tensorrt}/onnx_to_tensorrt.h (88%)
create mode 100644 src/operator/subgraph/tensorrt/tensorrt-inl.h
create mode 100644 src/operator/subgraph/tensorrt/tensorrt.cc
rename src/operator/{contrib => subgraph/tensorrt}/tensorrt.cu (69%)
delete mode 100644 tests/cpp/misc/serialization.cc
delete mode 100644 tests/python/tensorrt/test_cycle.py
delete mode 100644 tests/python/tensorrt/test_training_warning.py
