Browse Source

fix_l2norm_bug

tags/v1.1.1
bai-yangfan 5 years ago
parent
commit
71cf368fb3
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindspore/core/utils/check_convert_utils.cc

+ 4
- 1
mindspore/core/utils/check_convert_utils.cc View File

@@ -230,8 +230,11 @@ namespace {
typedef std::map<std::string, std::function<ValuePtr(ValuePtr)>> AttrFunction;

ValuePtr L2NormalizeAttrConversion(ValuePtr attr) {
if (attr->isa<Int64Imm>()) {
return attr;
}
auto attr_value = GetValue<std::vector<int64_t>>(attr);
return MakeValue<int64_t>(attr_value[0]);
return MakeValue(attr_value[0]);
}

std::map<std::string, AttrFunction> kIrAttrToOpAttr = {{"L2Normalize", {{"axis", L2NormalizeAttrConversion}}},


Loading…
Cancel
Save