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

add support for bool

This commit is contained in:
y00445136 2020-04-02 11:39:24 +08:00
parent cf54ecfe6e
commit b3cd639c09
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/utils/tensorprint_utils.cc
View File

@ -37,13 +37,13 @@ static std::map<std::string, TypeId> print_type_map = {
{"int32_t", TypeId::kNumberTypeInt32}, {"uint32_t", TypeId::kNumberTypeUInt32}, {"int32_t", TypeId::kNumberTypeInt32}, {"uint32_t", TypeId::kNumberTypeUInt32},
{"int64_t", TypeId::kNumberTypeInt64}, {"uint64_t", TypeId::kNumberTypeUInt64}, {"int64_t", TypeId::kNumberTypeInt64}, {"uint64_t", TypeId::kNumberTypeUInt64},
{"float16", TypeId::kNumberTypeFloat16}, {"float", TypeId::kNumberTypeFloat32}, {"float16", TypeId::kNumberTypeFloat16}, {"float", TypeId::kNumberTypeFloat32},
{"double", TypeId::kNumberTypeFloat64}}; {"double", TypeId::kNumberTypeFloat64}, {"bool", TypeId::kNumberTypeBool}};
static std::map<std::string, size_t> type_size_map = { static std::map<std::string, size_t> type_size_map = {
{"int8_t", sizeof(int8_t)}, {"uint8_t", sizeof(uint8_t)}, {"int16_t", sizeof(int16_t)}, {"int8_t", sizeof(int8_t)}, {"uint8_t", sizeof(uint8_t)}, {"int16_t", sizeof(int16_t)},
{"uint16_t", sizeof(uint16_t)}, {"int32_t", sizeof(int32_t)}, {"uint32_t", sizeof(uint32_t)}, {"uint16_t", sizeof(uint16_t)}, {"int32_t", sizeof(int32_t)}, {"uint32_t", sizeof(uint32_t)},
{"int64_t", sizeof(int64_t)}, {"uint64_t", sizeof(uint64_t)}, {"float16", sizeof(float) / 2}, {"int64_t", sizeof(int64_t)}, {"uint64_t", sizeof(uint64_t)}, {"float16", sizeof(float) / 2},
{"float", sizeof(float)}, {"double", sizeof(double)}}; {"float", sizeof(float)}, {"double", sizeof(double)}, {"bool", sizeof(bool)}};
bool ParseTensorShape(const std::string &input_shape_str, std::vector<int> *const tensor_shape, size_t *dims) { bool ParseTensorShape(const std::string &input_shape_str, std::vector<int> *const tensor_shape, size_t *dims) {
if (tensor_shape == nullptr) { if (tensor_shape == nullptr) {
@ -107,7 +107,7 @@ bool ConvertDataItem2Tensor(const std::vector<tdt::DataItem> &items) {
} else { } else {
auto type_iter = print_type_map.find(item.tensorType_); auto type_iter = print_type_map.find(item.tensorType_);
if (type_iter == print_type_map.end()) { if (type_iter == print_type_map.end()) {
MS_LOG(ERROR) << "type of tensor need to print is not soupport" << item.tensorType_; MS_LOG(ERROR) << "type of tensor need to print is not support " << item.tensorType_;
continue; continue;
} }
auto type_id = type_iter->second; auto type_id = type_iter->second;