mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!47900 bug fix 

Merge pull request !47900 from liubuyu/r2.0.0-alpha
This commit is contained in:
i-robot 2023-01-16 12:41:06 +00:00 committed by Gitee
parent 7179720c99 5fd15897cc
commit 0c04bdff44
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/transform/graph_ir/convert.cc
View File

@ -3020,6 +3020,8 @@ OpAdapterPtr GeOpConvertor::GetForReduceAdapter(const AnfNodePtr &node, const Op
auto op_name = common::AnfAlgo::GetCNodeName(node);
if (op_name == prim::kPrimReduceSum->name()) {
return FindAdapter("ReduceSumD", true);
} else if (op_name == "Conv3DTranspose") {
return FindAdapter("Conv3DTransposeD", true);
}
return adapter;
}

17
mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.cc
View File

@ -123,6 +123,21 @@ ATTR_MAP(Conv2DBackpropFilterD) = {
OUTPUT_MAP(Conv2DBackpropFilterD) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(Conv2DBackpropFilterD, kNameConv2DBackpropFilterD, ADPT_DESC(Conv2DBackpropFilterD))
// Conv3DTransposeD
INPUT_MAP(Conv3DTransposeD) = {
{1, INPUT_DESC(x)}, {2, INPUT_DESC(filter)}, {3, INPUT_DESC(bias)}, {4, INPUT_DESC(offset_w)}};
ATTR_MAP(Conv3DTransposeD) = {
{"input_size", ATTR_DESC(input_size, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())},
{"strides", ATTR_DESC(strides, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())},
{"pad_list", ATTR_DESC(pads, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())},
{"dilations", ATTR_DESC(dilations, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())},
{"groups", ATTR_DESC(groups, AnyTraits<int64_t>())},
{"format", ATTR_DESC(data_format, AnyTraits<std::string>())},
{"output_padding", ATTR_DESC(output_padding, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())},
};
OUTPUT_MAP(Conv3DTransposeD) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(Conv3DTransposeD, kNameConv3DTransposeD, ADPT_DESC(Conv3DTransposeD))
// Conv3DTranspose
INPUT_MAP(Conv3DTranspose) = {{1, INPUT_DESC(input_size)},
{2, INPUT_DESC(x)},
@ -139,7 +154,7 @@ ATTR_MAP(Conv3DTranspose) = {
{"offset_x", ATTR_DESC(offset_x, AnyTraits<int64_t>())}};
ATTR_INPUT_MAP(Conv3DTranspose) = {{"input_size", 1}};
OUTPUT_MAP(Conv3DTranspose) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(Conv3DTranspose, kNameConv3DTransposeD, ADPT_DESC(Conv3DTranspose))
REG_ADPT_DESC(Conv3DTranspose, "Conv3DTransposeD", ADPT_DESC(Conv3DTranspose))
// Conv3D
INPUT_MAP(Conv3D) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(filter)}, {3, INPUT_DESC(bias)}, {4, INPUT_DESC(offset_w)}};

3
mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.h
View File

@ -45,6 +45,9 @@ DECLARE_OP_USE_OUTPUT(Conv2DBackpropFilterD)
DECLARE_OP_ADAPTER(Conv3DTranspose)
DECLARE_OP_USE_OUTPUT(Conv3DTranspose)
DECLARE_OP_ADAPTER(Conv3DTransposeD)
DECLARE_OP_USE_OUTPUT(Conv3DTransposeD)
DECLARE_OP_ADAPTER(Conv3D)
DECLARE_OP_USE_OUTPUT(Conv3D)