diff --git a/lite/core/mir/memory_optimize_pass.cc b/lite/core/mir/memory_optimize_pass.cc index ef6dfcff0980b4831c446f7abe501a0752b26508..1f2355e8a3205cce3410bd2cb6ac4a17d8fde602 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"};