• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**
2  * Copyright 2019-2023 Huawei Technologies Co., Ltd
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #ifndef MINDSPORE_CORE_BASE_SPARSE_OP_NAME_H_
17 #define MINDSPORE_CORE_BASE_SPARSE_OP_NAME_H_
18 
19 namespace mindspore {
20 // Sparse ops
21 constexpr auto kSparseFillEmptyRowsGradOpName = "SparseFillEmptyRowsGrad";
22 constexpr auto kSparseSparseMinimumOpName = "SparseSparseMinimum";
23 constexpr auto kSparseCrossOpName = "SparseCross";
24 constexpr auto kRaggedTensorToTensorOpName = "RaggedTensorToTensor";
25 constexpr auto kSparseTensorDenseMatmulOpName = "SparseTensorDenseMatmul";
26 constexpr auto kSparseFillEmptyRowsOpName = "SparseFillEmptyRows";
27 constexpr auto kSparseToDenseV2OpName = "SparseToDenseV2";
28 constexpr auto kSparseSoftmaxOpName = "SparseSoftmax";
29 constexpr auto kSparseAddmmOpName = "SparseAddmm";
30 constexpr auto kSparseSparseMaximumOpName = "SparseSparseMaximum";
31 constexpr auto kCSRReduceSumOpName = "CSRReduceSum";
32 constexpr auto kCSRMVOpName = "CSRMV";
33 constexpr auto kCSRMMOpName = "CSRMM";
34 constexpr auto kCSRMulOpName = "CSRMul";
35 constexpr auto kCSRGatherOpName = "CSRGather";
36 constexpr auto kCSR2COOOpName = "CSR2COO";
37 constexpr auto kSparseDenseCwiseAddOpName = "SparseDenseCwiseAdd";
38 constexpr auto kSparseDenseCwiseDivOpName = "SparseDenseCwiseDiv";
39 constexpr auto kSparseDenseCwiseMulOpName = "SparseDenseCwiseMul";
40 constexpr auto kCOO2CSROpName = "COO2CSR";
41 constexpr auto kCSRDivOpName = "CSRDiv";
42 constexpr auto kDenseToDenseSetOperationOpName = "DenseToDenseSetOperation";
43 constexpr auto kSparseMatrixAddOpName = "SparseMatrixAdd";
44 constexpr auto kSparseMatrixMulOpName = "SparseMatrixMul";
45 constexpr auto kSparseAddOpName = "SparseAdd";
46 constexpr auto kSparseSegmentMeanGradOpName = "SparseSegmentMeanGrad";
47 constexpr auto kSparseSegmentMeanWithNumSegmentsOpName = "SparseSegmentMeanWithNumSegments";
48 constexpr auto kSparseConcatOpName = "SparseConcat";
49 constexpr auto kSparseMatrixNNZOpName = "SparseMatrixNNZ";
50 constexpr auto kSparseMatrixTransposeOpName = "SparseMatrixTranspose";
51 constexpr auto kSparseMatrixSoftmaxOpName = "SparseMatrixSoftmax";
52 constexpr auto kSparseMatrixMatMulOpName = "SparseMatrixMatMul";
53 constexpr auto kSparseMatrixSparseMatMulOpName = "SparseMatrixSparseMatMul";
54 constexpr auto kSparseMatrixOrderingAMDOpName = "SparseMatrixOrderingAMD";
55 constexpr auto kSparseSegmentSumOpName = "SparseSegmentSum";
56 constexpr auto kSparseSegmentSumGradOpName = "SparseSegmentSumGrad";
57 constexpr auto kSparseSegmentSumWithNumSegmentsOpName = "SparseSegmentSumWithNumSegments";
58 constexpr auto kSparseSegmentSqrtNOpName = "SparseSegmentSqrtN";
59 constexpr auto kSparseSegmentSqrtNGradOpName = "SparseSegmentSqrtNGrad";
60 constexpr auto kSparseSegmentSqrtNWithNumSegmentsOpName = "SparseSegmentSqrtNWithNumSegments";
61 constexpr auto kRaggedTensorToSparseOpName = "RaggedTensorToSparse";
62 constexpr auto kDenseToSparseSetOperationOpName = "DenseToSparseSetOperation";
63 constexpr auto kSparseTensorToCSRSparseMatrixOpName = "SparseTensorToCSRSparseMatrix";
64 constexpr auto kCSRSparseMatrixToSparseTensorOpName = "CSRSparseMatrixToSparseTensor";
65 constexpr auto kSparseSplitOpName = "SparseSplit";
66 constexpr auto kSparseReshapeOpName = "SparseReshape";
67 constexpr auto kSparseReorderOpName = "SparseReorder";
68 constexpr auto kSparseSegmentMeanOpName = "SparseSegmentMean";
69 
70 // Sparse Grad ops
71 constexpr auto kSparseAddGradOpName = "SparseAddGrad";
72 constexpr auto kSparseTensorDenseAddOpName = "SparseTensorDenseAdd";
73 constexpr auto kSparseSliceOpName = "SparseSlice";
74 constexpr auto kSparseSliceGradOpName = "SparseSliceGrad";
75 
76 // COOTensor
77 constexpr auto kMakeCOOTensorOpName = "MakeCOOTensor";
78 constexpr auto kCOOTensorGetValuesOpName = "COOTensorGetValues";
79 constexpr auto kCOOTensorGetIndicesOpName = "COOTensorGetIndices";
80 constexpr auto kCOOTensorGetDenseShapeOpName = "COOTensorGetDenseShape";
81 constexpr auto kCOOTensorDenseMatmulOpName = "COOTensorDenseMatmul";
82 
83 // RowTensor
84 constexpr auto kMakeRowTensorOpName = "MakeRowTensor";
85 constexpr auto kRowTensorGetValuesOpName = "RowTensorGetValues";
86 constexpr auto kRowTensorGetIndicesOpName = "RowTensorGetIndices";
87 constexpr auto kRowTensorGetDenseShapeOpName = "RowTensorGetDenseShape";
88 constexpr auto kRowTensorAddOpName = "RowTensorAdd";
89 
90 // CSRTensor
91 constexpr auto kMakeCSRTensorOpName = "MakeCSRTensor";
92 constexpr auto kCSRTensorGetValuesOpName = "CSRTensorGetValues";
93 constexpr auto kCSRTensorGetIndptrOpName = "CSRTensorGetIndptr";
94 constexpr auto kCSRTensorGetIndicesOpName = "CSRTensorGetIndices";
95 constexpr auto kCSRTensorGetDenseShapeOpName = "CSRTensorGetDenseShape";
96 constexpr auto kIsCSRFuncOpName = "IsCSRFunc";
97 
98 // MapTensor
99 constexpr auto kMakeMapParameterOpName = "MakeMapParameter";
100 constexpr auto kMapTensorGetOpName = "MapTensorGet";
101 constexpr auto kMapTensorPutOpName = "MapTensorPut";
102 constexpr auto kMapTensorEraseOpName = "MapTensorErase";
103 constexpr auto kMapTensorPutWithStatusOpName = "MapTensorPutWithStatus";
104 constexpr auto kMapTensorGetDefaultValueOpName = "MapTensorGetDefaultValue";
105 constexpr auto kMapTensorGetPermitFilterValueOpName = "MapTensorGetPermitFilterValue";
106 constexpr auto kMapTensorGetEvictFilterValueOpName = "MapTensorGetEvictFilterValue";
107 constexpr auto kMapTensorGetKeysOpName = "MapTensorGetKeys";
108 constexpr auto kMapTensorGetValuesOpName = "MapTensorGetValues";
109 constexpr auto kMapTensorGetDataOpName = "MapTensorGetData";
110 constexpr auto kMapTensorGetGradOpName = "MapTensorGetGrad";
111 constexpr auto kCSRSparseMatrixToDenseOpName = "CSRSparseMatrixToDense";
112 constexpr auto kSspaddmmOpName = "Sspaddmm";
113 constexpr auto kDenseToCSRSparseMatrixOpName = "DenseToCSRSparseMatrix";
114 constexpr auto kSparseToDenseOpName = "SparseToDense";
115 }  // namespace mindspore
116 
117 #endif  // MINDSPORE_CORE_BASE_SPARSE_OP_NAME_H_
118