Browse Source

!11223 [MS_LITE] female model step2

From: @YeFeng_24
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
9dd48d1e81
2 changed files with 5 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/test/models_tf.cfg
  2. +4
    -0
      mindspore/lite/tools/benchmark/benchmark.cc

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

@@ -2,3 +2,4 @@ decoder_step_201217.pb 5
decoder_step_201217_modified.pb 5 decoder_step_201217_modified.pb 5
unet_model_reconstruct.pb 1;1,256,256,3 unet_model_reconstruct.pb 1;1,256,256,3
encoder_201228.pb 3;1:1,22:1 encoder_201228.pb 3;1:1,22:1
female_model_step2_int16_noiseout.pb 66

+ 4
- 0
mindspore/lite/tools/benchmark/benchmark.cc View File

@@ -345,6 +345,10 @@ int Benchmark::CompareDataGetTotalBiasAndSize(const std::string &name, tensor::M
bias = CompareData<int32_t>(name, tensor->shape(), mutableData); bias = CompareData<int32_t>(name, tensor->shape(), mutableData);
break; break;
} }
case TypeId::kNumberTypeInt16: {
bias = CompareData<int16_t>(name, tensor->shape(), mutableData);
break;
}
default: default:
MS_LOG(ERROR) << "Datatype " << msCalibDataType << " is not supported."; MS_LOG(ERROR) << "Datatype " << msCalibDataType << " is not supported.";
return RET_ERROR; return RET_ERROR;


Loading…
Cancel
Save