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

change input_size of Conv3dTranpose from INPUT_ATTR_MAP to ATTR_MAP.

This commit is contained in:
wangshuide2020 2021-07-07 16:04:51 +08:00
parent b4ed7172dc
commit 3bc7e787d0
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -68,9 +68,8 @@ REG_ADPT_DESC(Conv2DBackpropFilterD, prim::kPrimConv2DBackpropFilter->name(), AD
// Conv3DTransposeD
INPUT_MAP(Conv3DTransposeD) = {
{1, INPUT_DESC(x)}, {2, INPUT_DESC(filter)}, {3, INPUT_DESC(bias)}, {4, INPUT_DESC(offset_w)}};
INPUT_ATTR_MAP(Conv3DTransposeD) = {
{5, ATTR_DESC(input_size, AnyTraits<std::vector<int64_t>>(), AnyTraits<std::vector<int64_t>>())}};
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>>())},

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

@ -42,7 +42,6 @@ DECLARE_OP_USE_OUTPUT(Conv2DBackpropFilterD)
DECLARE_OP_ADAPTER(Conv3DTransposeD)
DECLARE_OP_USE_ENUM(Conv3DTransposeD)
DECLARE_OP_USE_INPUT_ATTR(Conv3DTransposeD)
DECLARE_OP_USE_OUTPUT(Conv3DTransposeD)
DECLARE_OP_ADAPTER(Conv3D)