From 21773b6ccd0b327d42feb15c4fe4e2fa4abb77cb Mon Sep 17 00:00:00 2001 From: Wilber Date: Sat, 21 Sep 2019 11:05:57 +0800 Subject: [PATCH] add yolo_box to mem_optimize_pass's unuse op set --- lite/core/mir/memory_optimize_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/lite/core/mir/memory_optimize_pass.cc b/lite/core/mir/memory_optimize_pass.cc index ef6dfcff09..1f2355e8a3 100644 --- a/lite/core/mir/memory_optimize_pass.cc +++ b/lite/core/mir/memory_optimize_pass.cc @@ -49,6 +49,7 @@ void MemoryOptimizePass::CollectLifeCycleByDevice( "equal", "lod_reset", "concat", + "yolo_box", "graph_op", "feed", "fetch"}; -- GitLab