Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_segment_mean.h37 …SparseSegmentMean() : BaseOperator(kNameSparseSegmentMean) { InitIOName({"x", "indices", "segment_… in SparseSegmentMean() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsparse_segment_mean_impl.cu97 cudaError_t SparseSegmentMean(const DataType *x_ptr, const IndexType *indices_ptr, const IndexType … in SparseSegmentMean() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dmath_ops.py4475 class SparseSegmentMean(Primitive): class