diff --git a/src/executor/trt_graph_executor.cc b/src/executor/trt_graph_executor.cc index 65dbb29792e0..ec35fee98a96 100644 --- a/src/executor/trt_graph_executor.cc +++ b/src/executor/trt_graph_executor.cc @@ -407,14 +407,7 @@ nnvm::Symbol TrtGraphExecutor::GetOptimizedSymbol() { Symbol ret; ret.outputs = std::vector(graph_.outputs.begin(), graph_.outputs.begin() + num_forward_outputs_); - ret = ret.Copy(); - static const Op* trt_op = Op::Get("_trt_op"); - DFSVisit(ret.outputs, [](const nnvm::NodePtr n) { - if (n->op() == trt_op) { - n->attrs.dict.clear(); - } - }); - return ret; + return ret.Copy(); } Executor *TrtGraphExecutor::TensorRTBind(nnvm::Symbol symbol,