diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/split_combination_ops_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare/split_combination_ops_declare.cc index ae39b1e629212d40156eb837ec7175184e0d3711..2b38364e0b56702ea259aff85c5b5dc18ccd2422 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/split_combination_ops_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/split_combination_ops_declare.cc @@ -25,14 +25,14 @@ INPUT_ATTR_MAP(SplitD) = {{2, ATTR_DESC(split_dim, AnyTraits())}, {3, ATTR_DESC(num_split, AnyTraits())}}; ATTR_MAP(SplitD) = EMPTY_ATTR_MAP; DYN_OUTPUT_MAP(SplitD) = {{0, DYN_OUTPUT_DESC(y)}}; -REG_ADPT_DESC(Split, kNameSplit, ADPT_DESC(SplitD)) +REG_ADPT_DESC(SplitD, kSplitDOpName, ADPT_DESC(SplitD)) // Split -INPUT_MAP(Split) = {{1, INPUT_DESC(split_dim)}, {2, INPUT_DESC(x)}}; -ATTR_INPUT_MAP(Split) = {{"axis", "split_dim"}}; -ATTR_MAP(Split) = {{"output_num", ATTR_DESC(num_split, AnyTraits())}}; +INPUT_MAP(Split) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(split_dim)}}; +INPUT_ATTR_MAP(Split) = {{3, ATTR_DESC(num_split, AnyTraits())}}; +ATTR_MAP(Split) = EMPTY_ATTR_MAP; DYN_OUTPUT_MAP(Split) = {{0, DYN_OUTPUT_DESC(y)}}; -REG_ADPT_DESC(SplitD, kSplitDOpName, ADPT_DESC(Split)) +REG_ADPT_DESC(Split, kNameSplit, ADPT_DESC(Split)) // Pack INPUT_MAP(Pack) = EMPTY_INPUT_MAP;