|
|
|
@@ -14,8 +14,8 @@ |
|
|
|
* limitations under the License. |
|
|
|
*/ |
|
|
|
|
|
|
|
#include "transform/graph_ir/op_declare/selection_ops_declare.h" |
|
|
|
#include <vector> |
|
|
|
#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<int>(), AnyTraits<std::vector<int64_t>>())}}; |
|
|
|
OUTPUT_MAP(ReverseV2D) = {{0, OUTPUT_DESC(y)}}; |
|
|
|
REG_ADPT_DESC(ReverseV2D, kNameReverseV2, ADPT_DESC(ReverseV2D)) |
|
|
|
} // namespace mindspore::transform |