Browse Source

single op parser add check for json input

pull/580/head
wxl 5 years ago
parent
commit
9869a2ea84
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      ge/offline/single_op_parser.cc

+ 4
- 4
ge/offline/single_op_parser.cc View File

@@ -191,14 +191,14 @@ void from_json(const Json &j, SingleOpTensorDesc &desc) {
}
string format_str = j.at(kKeyFormat).get<string>();
string type_str = j.at(kKeyType).get<string>();
is_tensor_valid &&= ge::TypeUtils::IsFormatValid(format_str);
is_tensor_valid &&= ge::TypeUtils::IsDataTypeValid(type_str);
is_tensor_valid = is_tensor_valid && ge::TypeUtils::IsFormatValid(format_str);
is_tensor_valid = is_tensor_valid && ge::TypeUtils::IsDataTypeValid(type_str);
desc.format = GetValue(kFormatDict, format_str, FORMAT_RESERVED);
desc.type = GetValue(kDataTypeDict, type_str, DT_UNDEFINED);
it = j.find(kKeyOriginFormat);
if (it != j.end()) {
string origin_format_str = j.at(kKeyOriginFormat).get<string>();
is_tensor_valid &&= ge::TypeUtils::IsFormatValid(origin_format_str);
is_tensor_valid = is_tensor_valid && ge::TypeUtils::IsFormatValid(origin_format_str);
desc.ori_format = GetValue(kFormatDict, origin_format_str, FORMAT_RESERVED);
}
auto tensor_name = j.find(kKeyName);
@@ -210,7 +210,7 @@ void from_json(const Json &j, SingleOpTensorDesc &desc) {
desc.dynamic_input_name = dynamic_input_name->get<string>();
}
if (!is_tensor_valid) {
ge::AttrUtils::SetBool(&desc, kSingleOpTensorDescValid, is_tensor_valid);
ge::AttrUtils::SetBool(desc, kSingleOpTensorDescValid, is_tensor_valid);
}
}



Loading…
Cancel
Save