diff --git a/mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc b/mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc index 2e955e7ca7..974d9a9f3e 100644 --- a/mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc +++ b/mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc @@ -23,9 +23,9 @@ namespace mindspore::lite { using kernel::KERNEL_ARCH::kNPU; enum InsertState { InsertNone, PreInsert, PostInsert, BothInsert }; std::set npu_insert_nodes = { - schema::PrimitiveType_Concat, schema::PrimitiveType_AddFusion, schema::PrimitiveType_Eltwise, - schema::PrimitiveType_Activation, schema::PrimitiveType_Split, schema::PrimitiveType_PadFusion, - schema::PrimitiveType_StridedSlice, schema::PrimitiveType_Activation}; + schema::PrimitiveType_Concat, schema::PrimitiveType_AddFusion, schema::PrimitiveType_Eltwise, + schema::PrimitiveType_Activation, schema::PrimitiveType_Split, schema::PrimitiveType_PadFusion, + schema::PrimitiveType_StridedSlice}; // this pass goal is to minimize subgraphs generated // by inserting nchw2nhwc or nhwc2nchw before or after the operator (e.g. concat, add, etc..) together with