Browse Source

remove dup act npu pass

pull/14202/head
zhaozhenlong 4 years ago
parent
commit
6ee5bee606
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc

+ 3
- 3
mindspore/lite/src/runtime/agent/npu/optimizer/npu_insert_transform_pass.cc View File

@@ -23,9 +23,9 @@ namespace mindspore::lite {
using kernel::KERNEL_ARCH::kNPU; using kernel::KERNEL_ARCH::kNPU;
enum InsertState { InsertNone, PreInsert, PostInsert, BothInsert }; enum InsertState { InsertNone, PreInsert, PostInsert, BothInsert };
std::set<mindspore::schema::PrimitiveType> npu_insert_nodes = { std::set<mindspore::schema::PrimitiveType> 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 // 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 // by inserting nchw2nhwc or nhwc2nchw before or after the operator (e.g. concat, add, etc..) together with


Loading…
Cancel
Save