Browse Source

add new model to the entrance guard

tags/v1.1.0
zengxianglong 5 years ago
parent
commit
bc2f6ee388
3 changed files with 3 additions and 2 deletions
  1. +0
    -1
      mindspore/lite/test/models_only_for_process.cfg
  2. +1
    -0
      mindspore/lite/test/models_tflite_awaretraining.cfg
  3. +2
    -1
      mindspore/lite/tools/common/node_util.cc

+ 0
- 1
mindspore/lite/test/models_only_for_process.cfg View File

@@ -12,7 +12,6 @@ lite-model_east-text-detector_dr_1.tflite
magenta_arbitrary-image-stylization-v1-256_fp16_transfer_1.tflite
magenta_arbitrary-image-stylization-v1-256_int8_transfer_1.tflite
magenta_arbitrary-image-stylization-v1-256_int8_prediction_1.tflite
gts_model_quant_frozen.tflite
albert_lite_base_squadv1_1.tflite
efficientnet_lite0_int8_2.tflite
efficientnet_lite1_int8_2.tflite


+ 1
- 0
mindspore/lite/test/models_tflite_awaretraining.cfg View File

@@ -37,3 +37,4 @@ detect.tflite
ssd_mobilenet_v1_1_default_1.tflite
object_detection_mobile_object_localizer_v1_1_default_1.tflite
gts_detect_0730_quant_frozen.tflite
gts_model_quant_frozen.tflite

+ 2
- 1
mindspore/lite/tools/common/node_util.cc View File

@@ -136,7 +136,8 @@ static const std::vector<schema::PrimitiveType> int8OpList = {schema::PrimitiveT
schema::PrimitiveType_DetectionPostProcess,
schema::PrimitiveType_Crop,
schema::PrimitiveType_PriorBox,
schema::PrimitiveType_QuantDTypeCast};
schema::PrimitiveType_QuantDTypeCast,
schema::PrimitiveType_L2Norm};

static const std::vector<schema::PrimitiveType> needInsertOpList = {
schema::PrimitiveType_Eltwise, schema::PrimitiveType_Activation, schema::PrimitiveType_Concat,


Loading…
Cancel
Save