未验证 提交 0be4ef9f 编写于 作者: H hong19860320 提交者: GitHub

[Core] Fix the missing of the attr name of the output scale (#4334)

上级 27150a06
......@@ -322,6 +322,7 @@ std::vector<float> OpInfo::GetOutputScale(const std::string &name,
int index;
CHECK(GetOutputArgname(name, &argname));
CHECK(GetOutputIndex(name, &index));
scale_name = argname + to_string(index) + "_scale";
}
return GetAttr<std::vector<float>>(scale_name);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册