From 1f151fbbc00751a590ec2377377003b421c45dec Mon Sep 17 00:00:00 2001 From: Caenorst Date: Mon, 7 Jan 2019 21:10:16 -0500 Subject: [PATCH] remove attributs clear on TRT nodes for GetOptimizedSymbol (#13703) --- src/executor/trt_graph_executor.cc | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) 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,