未验证 提交 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 { ...@@ -140,12 +140,14 @@ class ExpandV2OpMaker : public framework::OpProtoAndCheckerMaker {
.SetDefault({}); .SetDefault({});
AddAttr<bool>("use_mkldnn", AddAttr<bool>("use_mkldnn",
"(bool, default false) Only used in mkldnn kernel") "(bool, default false) Only used in mkldnn kernel")
.SetDefault(false); .SetDefault(false)
.AsExtra();
AddAttr<std::string>( AddAttr<std::string>(
"mkldnn_data_type", "mkldnn_data_type",
"(string, default \"float32\"). Data type of mkldnn kernel") "(string, default \"float32\"). Data type of mkldnn kernel")
.SetDefault("float32") .SetDefault("float32")
.InEnum({"float32", "bfloat16"}); .InEnum({"float32", "bfloat16"})
.AsExtra();
AddComment(R"DOC( AddComment(R"DOC(
Expand the input to the given shape. The rank of X 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. 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.
先完成此消息的编辑!
想要评论请 注册