diff --git a/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc b/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc index 19d0f26eae66..4f80d277cad8 100644 --- a/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc +++ b/src/operator/subgraph/tensorrt/nnvm_to_onnx.cc @@ -31,7 +31,6 @@ #include #include #include -#include #include "../../../common/utils.h" #include "../../../ndarray/ndarray_function.h" @@ -39,6 +38,7 @@ #include "../../nn/activation-inl.h" #include "../../nn/batch_norm-inl.h" #include "../../nn/convolution-inl.h" +#include "../../nn/deconvolution-inl.h" #include "../../nn/fully_connected-inl.h" #include "../../nn/pooling-inl.h" #include "../../nn/concat-inl.h"