Home
last modified time | relevance | path

Searched defs:SparseMatrixSoftmax (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_matrix_softmax.h34 SparseMatrixSoftmax() : BaseOperator(kNameSparseMatrixSoftmax) { in SparseMatrixSoftmax() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsparse_matrix_softmax_impl.cu57 cudaError_t SparseMatrixSoftmax(int shape_size, int batch_pointers_size, int row_pointers_size, int… in SparseMatrixSoftmax() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dsparse_ops.py1562 class SparseMatrixSoftmax(Primitive): class