未验证 提交 9c9eba13 编写于 作者: J Jiawei Wang 提交者: GitHub

fix extra op for expand, expand_as, tile, unstack (#35598)

* fix extra op for expand, expand_as, tile, unstack

* Update expand_v2_op.cc
上级 d7985052
......@@ -140,12 +140,14 @@ class ExpandV2OpMaker : public framework::OpProtoAndCheckerMaker {
.SetDefault({});
AddAttr<bool>("use_mkldnn",
"(bool, default false) Only used in mkldnn kernel")
.SetDefault(false);
.SetDefault(false)
.AsExtra();
AddAttr<std::string>(
"mkldnn_data_type",
"(string, default \"float32\"). Data type of mkldnn kernel")
.SetDefault("float32")
.InEnum({"float32", "bfloat16"});
.InEnum({"float32", "bfloat16"})
.AsExtra();
AddComment(R"DOC(
Expand the input to the given shape. The rank of X
should be in [1, 6] and size of 'shape' must be in [1, 6] also.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册