diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.cc index 78a6ce00844..837955a1a03 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.cc @@ -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>(), AnyTraits>())}}; ATTR_MAP(Conv3DTransposeD) = { + {"input_size", ATTR_DESC(input_size, AnyTraits>(), AnyTraits>())}, {"strides", ATTR_DESC(strides, AnyTraits>(), AnyTraits>())}, {"pad_list", ATTR_DESC(pads, AnyTraits>(), AnyTraits>())}, {"dilations", ATTR_DESC(dilations, AnyTraits>(), AnyTraits>())}, diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.h b/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.h index d7fe8a2039f..4abdc40e6a9 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.h +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/nn_calculation_ops_declare.h @@ -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)