未验证 提交 b3f60f39 编写于 作者: 傅剑寒 提交者: GitHub

fix set_value_65965 (#50340)

上级 197a4ffe
...@@ -194,6 +194,19 @@ KernelSignature SetValueOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -194,6 +194,19 @@ KernelSignature SetValueOpArgumentMapping(const ArgumentMappingContext& ctx) {
"shape", "shape",
"bool_values"}, "bool_values"},
{"Out"}); {"Out"});
} else if (ctx.HasAttr("fp16_values")) {
// NOTE(LiuYang):Here any_cast doesn't support fp16 values.
return KernelSignature("set_value",
{"Input"},
{"StartsTensorList",
"EndsTensorList",
"steps",
"axes",
"decrease_axes",
"none_axes",
"shape",
"fp16_values"},
{"Out"});
} }
} }
} else { } else {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册