From 7e74b8583db9fbca749cf271ad6d99a9d9b43d20 Mon Sep 17 00:00:00 2001 From: z00621985 Date: Wed, 1 May 2024 16:42:27 +0800 Subject: [PATCH] [MS][OP] change split adapter from splitD to split --- .../op_declare/split_combination_ops_declare.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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 ae39b1e62921..2b38364e0b56 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; -- Gitee