Browse Source

2

tags/v1.2.0-rc1
yefeng 5 years ago
parent
commit
e4657e9889
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
unet_model_reconstruct.pb 1;1,256,256,3
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);
break;
}
case TypeId::kNumberTypeInt16: {
bias = CompareData<int16_t>(name, tensor->shape(), mutableData);
break;
}
default:
MS_LOG(ERROR) << "Datatype " << msCalibDataType << " is not supported.";
return RET_ERROR;


Loading…
Cancel
Save