From e4657e9889fbd5fb173a0e1c3ff2564b9d165564 Mon Sep 17 00:00:00 2001
From: yefeng <yefeng24@huawei.com>
Date: Wed, 13 Jan 2021 14:11:46 +0800
Subject: [PATCH] 2

---
 mindspore/lite/test/models_tf.cfg           | 1 +
 mindspore/lite/tools/benchmark/benchmark.cc | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/mindspore/lite/test/models_tf.cfg b/mindspore/lite/test/models_tf.cfg
index e62f926486..b5377cbf6b 100644
--- a/mindspore/lite/test/models_tf.cfg
+++ b/mindspore/lite/test/models_tf.cfg
@@ -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
diff --git a/mindspore/lite/tools/benchmark/benchmark.cc b/mindspore/lite/tools/benchmark/benchmark.cc
index 82723d216b..2ab2d50280 100644
--- a/mindspore/lite/tools/benchmark/benchmark.cc
+++ b/mindspore/lite/tools/benchmark/benchmark.cc
@@ -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;