Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48374 [MS][LITE][r1.8] schema tune r1.8 

Merge pull request !48374 from Greatpan/schema_tune_r1.8
This commit is contained in:
i-robot 2023-02-03 10:06:28 +00:00 committed by Gitee
parent 6878b3d26f 66f25f7f49
commit a6dfb0d47e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
6 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/op_base.h
View File

@ -521,9 +521,10 @@ enum PrimType {
PrimType_GatherD = 210,
PrimType_GroupNormFusion = 211,
PrimType_Log1p = 212,
PrimType_SparseFillEmptyRows = 213,
PrimType_SparseReshape = 214,
PrimType_SparseSegmentSum = 215,
PrimType_TensorScatterAdd = 213,
PrimType_SparseFillEmptyRows = 214,
PrimType_SparseReshape = 215,
PrimType_SparseSegmentSum = 216,
PrimType_MIN = PrimType_NONE,
PrimType_MAX = PrimType_SparseSegmentSum + 1,

4
mindspore/lite/schema/ops.fbs
View File

@ -230,6 +230,7 @@ union PrimitiveType {
GatherD,
GroupNormFusion,
Log1p,
TensorScatterAdd,
SparseFillEmptyRows,
SparseReshape,
SparseSegmentSum,
@ -1292,6 +1293,9 @@ table GroupNormFusion {
table Log1p {
}
table TensorScatterAdd {
}
table SparseFillEmptyRows {
}

4
mindspore/lite/src/common/ops/ops_def.cc
View File

@ -230,6 +230,7 @@ OP_TYPE(FormatTranspose)
OP_TYPE(GatherD)
OP_TYPE(GroupNormFusion)
OP_TYPE(Log1p)
OP_TYPE(TensorScatterAdd)
OP_TYPE(SparseFillEmptyRows)
OP_TYPE(SparseReshape)
OP_TYPE(SparseSegmentSum)
@ -1292,6 +1293,9 @@ OP_SCHEMA_DEF_END(GroupNormFusion)
OP_SCHEMA_DEF(Log1p)
OP_SCHEMA_DEF_END(Log1p)
OP_SCHEMA_DEF(TensorScatterAdd)
OP_SCHEMA_DEF_END(TensorScatterAdd)
OP_SCHEMA_DEF(SparseFillEmptyRows)
OP_SCHEMA_DEF_END(SparseFillEmptyRows)

2
mindspore/lite/src/common/ops/ops_func_declare.h
View File

@ -259,6 +259,7 @@
#include "ops/grad/nllloss_grad.h"
#include "ops/format_transpose.h"
#include "ops/gather_d.h"
#include "ops/tensor_scatter_add.h"
#include "ops/sparse_fill_empty_rows.h"
#include "ops/sparse_reshape.h"
#include "ops/sparse_segment_sum.h"
@ -489,6 +490,7 @@ FUNC_MSOP2SCHEMAOP_DECLARE(FormatTranspose)
FUNC_MSOP2SCHEMAOP_DECLARE(GatherD)
FUNC_MSOP2SCHEMAOP_DECLARE(GroupNormFusion)
FUNC_MSOP2SCHEMAOP_DECLARE(Log1p)
FUNC_MSOP2SCHEMAOP_DECLARE(TensorScatterAdd)
FUNC_MSOP2SCHEMAOP_DECLARE(SparseFillEmptyRows)
FUNC_MSOP2SCHEMAOP_DECLARE(SparseReshape)
FUNC_MSOP2SCHEMAOP_DECLARE(SparseSegmentSum)

2
mindspore/lite/test/config_level0/cropped_size.cfg
View File

@ -1,2 +1,2 @@
Note: This is the mindspore Lite inference framework size threshold. Offline review is required before modify this value!!!
1039556
1042760

2
mindspore/lite/test/config_level1/cropped_size.cfg
View File

@ -1,2 +1,2 @@
Note: This is the mindspore Lite inference framework size threshold. Modifying this threshold requires meeting review.
1039556
1042760