diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc index 594e050ee57..bba566a1da1 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc @@ -39,12 +39,10 @@ ATTR_MAP(CumprodD) = {{"exclusive", ATTR_DESC(exclusive, AnyTraits())}, OUTPUT_MAP(CumprodD) = {{0, OUTPUT_DESC(y)}}; REG_ADPT_DESC(CumprodD, kNameCumProd, ADPT_DESC(CumprodD)) -INPUT_MAP(SliceD) = {{1, INPUT_DESC(x)}}; -INPUT_ATTR_MAP(SliceD) = {{2, ATTR_DESC(offsets, AnyTraits(), AnyTraits>())}, - {3, ATTR_DESC(size, AnyTraits(), AnyTraits>())}}; -ATTR_MAP(SliceD) = EMPTY_ATTR_MAP; -OUTPUT_MAP(SliceD) = {{0, OUTPUT_DESC(y)}}; -REG_ADPT_DESC(SliceD, kNameSlice, ADPT_DESC(SliceD)) +INPUT_MAP(Slice) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(offsets)}, {3, INPUT_DESC(size)}}; +ATTR_MAP(Slice) = EMPTY_ATTR_MAP; +OUTPUT_MAP(Slice) = {{0, OUTPUT_DESC(y)}}; +REG_ADPT_DESC(Slice, kNameSlice, ADPT_DESC(Slice)) // TopK INPUT_MAP(TopK) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(k)}}; diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h index 1e74469678b..63f6644922d 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h @@ -23,9 +23,8 @@ #include "ops/selection_ops.h" namespace mindspore::transform { -DECLARE_OP_ADAPTER(SliceD) -DECLARE_OP_USE_INPUT_ATTR(SliceD) -DECLARE_OP_USE_OUTPUT(SliceD) +DECLARE_OP_ADAPTER(Slice) +DECLARE_OP_USE_OUTPUT(Slice) DECLARE_OP_ADAPTER(ScatterNd) DECLARE_OP_USE_OUTPUT(ScatterNd)