From 8093f770bfec771d35553ce4ba8adcfaa0ff655e Mon Sep 17 00:00:00 2001 From: liangchenghui Date: Tue, 27 Oct 2020 20:00:31 +0800 Subject: [PATCH] Add ReverseV2 op for old backend. --- mindspore/ccsrc/transform/graph_ir/op_adapter_map.h | 1 + .../graph_ir/op_declare/selection_ops_declare.cc | 8 +++++++- .../transform/graph_ir/op_declare/selection_ops_declare.h | 3 +++ 3 files changed, 11 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/transform/graph_ir/op_adapter_map.h b/mindspore/ccsrc/transform/graph_ir/op_adapter_map.h index 189e57ad8a..e375fa9ff7 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_adapter_map.h +++ b/mindspore/ccsrc/transform/graph_ir/op_adapter_map.h @@ -199,6 +199,7 @@ constexpr const char kNameEditDistance[] = "EditDistance"; constexpr const char kNameCase[] = "Case"; constexpr const char kNameAssert[] = "Assert"; constexpr const char kNameCTCGreedyDecoder[] = "CTCGreedyDecoder"; +constexpr const char kNameReverseV2[] = "ReverseV2"; class OpAdapterMap { public: 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 27dfe612b6..a6431fdfd1 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 @@ -14,8 +14,8 @@ * limitations under the License. */ -#include "transform/graph_ir/op_declare/selection_ops_declare.h" #include +#include "transform/graph_ir/op_declare/selection_ops_declare.h" namespace mindspore::transform { // CumsumD @@ -133,4 +133,10 @@ INPUT_MAP(UnsortedSegmentMin) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids) ATTR_MAP(UnsortedSegmentMin) = EMPTY_ATTR_MAP; OUTPUT_MAP(UnsortedSegmentMin) = {{0, OUTPUT_DESC(y)}}; REG_ADPT_DESC(UnsortedSegmentMin, prim::kPrimUnsortedSegmentMin->name(), ADPT_DESC(UnsortedSegmentMin)) + +// ReverseV2 +INPUT_MAP(ReverseV2D) = {{1, INPUT_DESC(x)}}; +ATTR_MAP(ReverseV2D) = {{"axis", ATTR_DESC(axis, AnyTraits(), AnyTraits>())}}; +OUTPUT_MAP(ReverseV2D) = {{0, OUTPUT_DESC(y)}}; +REG_ADPT_DESC(ReverseV2D, kNameReverseV2, ADPT_DESC(ReverseV2D)) } // namespace mindspore::transform 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 c2b3097738..cd207c73f6 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 @@ -77,5 +77,8 @@ DECLARE_OP_USE_OUTPUT(CumsumD) DECLARE_OP_ADAPTER(GatherV2) DECLARE_OP_USE_OUTPUT(GatherV2) + +DECLARE_OP_ADAPTER(ReverseV2D) +DECLARE_OP_USE_OUTPUT(ReverseV2D) } // namespace mindspore::transform #endif // MINDSPORE_CCSRC_TRANSFORM_GRAPH_IR_OP_DECLARE_SELECTION_OPS_DECLARE_H_