Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48744 Fix Attr Bug of SpaceToDepth and DepthToSpace 

Merge pull request !48744 from jiaorui/fix-acl
This commit is contained in:
i-robot 2023-02-11 12:33:34 +00:00 committed by Gitee
parent 1d2710aa07 1d2c45b155
commit c29a919b72
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/transform/graph_ir/op_declare/transformation_ops_declare.cc
View File

@ -70,7 +70,8 @@ REG_ADPT_DESC(TransposeD, prim::kPrimTransposeD->name(), ADPT_DESC(Transpose))
// SpaceToDepth
INPUT_MAP(SpaceToDepth) = {{1, INPUT_DESC(x)}};
ATTR_MAP(SpaceToDepth) = {{"block_size", ATTR_DESC(block_size, AnyTraits<int64_t>())}};
ATTR_MAP(SpaceToDepth) = {{"block_size", ATTR_DESC(block_size, AnyTraits<int64_t>())},
{"data_format", ATTR_DESC(data_format, AnyTraits<std::string>())}};
OUTPUT_MAP(SpaceToDepth) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(SpaceToDepth, kNameSpaceToDepth, ADPT_DESC(SpaceToDepth))

4
mindspore/core/ops/depth_to_space.cc
View File

@ -65,6 +65,10 @@ abstract::ShapePtr DepthToSpaceInferShape(const PrimitivePtr &primitive,
auto data_format_ptr = primitive->GetAttr("format");
int64_t format = CheckAndConvertUtils::GetAndCheckFormat(primitive->GetAttr("format"));
(void)primitive->AddAttr("data_format", data_format_ptr);
// Set default value for mode
if (!primitive->GetAttr("mode")) {
(void)primitive->AddAttr("mode", MakeValue("DCR"));
}
const int64_t dim_0 = 0;
const int64_t dim_1 = 1;
const int64_t dim_2 = 2;