!28363 add GatherD ir

Merge pull request !28363 from yanzhenxiang2020/add_gatherd_ir

mindspore/ccsrc/include/transform/graph_ir/op_adapter_map.h

@@ -118,6 +118,7 @@ constexpr const char kNamePadV1[] = "PadV1";
 constexpr const char kNameMirrorPad[] = "MirrorPad";
 constexpr const char kNameMirrorPadGrad[] = "MirrorPadGrad";
 constexpr const char kNameGatherNd[] = "GatherNd";
+constexpr const char kNameGatherD[] = "GatherD";
 constexpr const char kNameArgmax[] = "Argmax";
 constexpr const char kNameArgmin[] = "Argmin";
 constexpr const char kNameArgMaxWithValue[] = "ArgMaxWithValue";

mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc

@@ -99,6 +99,12 @@ ATTR_MAP(GatherNd) = EMPTY_ATTR_MAP;
 OUTPUT_MAP(GatherNd) = {{0, OUTPUT_DESC(y)}};
 REG_ADPT_DESC(GatherNd, kNameGatherNd, ADPT_DESC(GatherNd))
 
+// GatherD
+INPUT_MAP(GatherD) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(dim)}, {3, INPUT_DESC(index)}};
+ATTR_MAP(GatherD) = EMPTY_ATTR_MAP;
+OUTPUT_MAP(GatherD) = {{0, OUTPUT_DESC(y)}};
+REG_ADPT_DESC(GatherD, kNameGatherD, ADPT_DESC(GatherD))
+
 // Range
 INPUT_MAP(RangeD) = {{1, INPUT_DESC(x)}};
 ATTR_MAP(RangeD) = {{"start", ATTR_DESC(start, AnyTraits<float>())},

mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h

@@ -37,6 +37,9 @@ DECLARE_OP_USE_OUTPUT(ScatterNonAliasingAdd)
 DECLARE_OP_ADAPTER(GatherNd)
 DECLARE_OP_USE_OUTPUT(GatherNd)
 
+DECLARE_OP_ADAPTER(GatherD)
+DECLARE_OP_USE_OUTPUT(GatherD)
+
 DECLARE_OP_ADAPTER(TopK)
 DECLARE_OP_USE_OUTPUT(TopK)