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

output tensor name

This commit is contained in:
yefeng 2021-09-02 21:04:58 +08:00
parent a28e423a10
commit a3a67e692f
1 changed files with 2 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
mindspore/lite/tools/benchmark/benchmark_base.cc
View File

@ -132,54 +132,13 @@ int BenchmarkBase::LoadInput() {
// calibData is FP32
int BenchmarkBase::ReadCalibData() {
auto new_file = flags_->benchmark_data_file_ + ".new";
const char *calib_data_path = new_file.c_str();
const char *calib_data_path = flags_->benchmark_data_file_.c_str();
// read calib data
std::ifstream in_file(calib_data_path);
if (!in_file.good()) {
std::cerr << "file: " << calib_data_path << " is not exist" << std::endl;
MS_LOG(ERROR) << "file: " << calib_data_path << " is not exist";
auto old_file = flags_->benchmark_data_file_;
const char *old_calib_data_path = old_file.c_str();
std::ifstream old_in_file(old_calib_data_path);
if (!old_in_file.good()) {
std::cerr << "file: " << old_calib_data_path << " is not exist" << std::endl;
MS_LOG(ERROR) << "file: " << old_calib_data_path << " is not exist";
return RET_ERROR;
}
if (!old_in_file.is_open()) {
std::cerr << "file: " << old_calib_data_path << " open failed" << std::endl;
MS_LOG(ERROR) << "file: " << old_calib_data_path << " open failed";
old_in_file.close();
return RET_ERROR;
}
MS_LOG(INFO) << "Start reading calibData file";
std::string line;
std::string tensor_name;
while (!old_in_file.eof()) {
getline(old_in_file, line);
std::stringstream string_line1(line);
size_t dim = 0;
string_line1 >> tensor_name >> dim;
std::vector<size_t> dims;
for (size_t i = 0; i < dim; i++) {
size_t tmp_dim;
string_line1 >> tmp_dim;
dims.push_back(tmp_dim);
}
auto ret = ReadTensorData(old_in_file, tensor_name, dims);
if (ret != RET_OK) {
MS_LOG(ERROR) << "Read tensor data failed, tensor name: " << tensor_name;
return RET_ERROR;
}
}
old_in_file.close();
MS_LOG(INFO) << "Finish reading calibData file";
return RET_OK;
return RET_ERROR;
}
if (!in_file.is_open()) {