Browse Source

!14056 tensorprint segmentation

From: @yepei6
Reviewed-by: @kingxian,@kisnwang
Signed-off-by: @kingxian
pull/14056/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
2d73a35793
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/utils/tensorprint_utils.cc

+ 1
- 0
mindspore/ccsrc/utils/tensorprint_utils.cc View File

@@ -256,6 +256,7 @@ bool SaveDataset2File(acltdtDataset *acl_dataset, const std::string &print_file_
}

tensor->set_tensor_type(parse_type);
acl_data = const_cast<char *>(reinterpret_cast<std::string *>(acl_data)->c_str());
std::string data(reinterpret_cast<const char *>(acl_data), acl_data_size);
tensor->set_tensor_content(data);
}


Loading…
Cancel
Save