| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_segment_impl.cu | 38 __global__ void SparseSegmentSumKernel(const R *x_ptr, const S *indices_ptr, const size_t *segment_… in SparseSegmentSumKernel()
66 __global__ void SparseSegmentSumKernel(const float *x_ptr, const S *indices_ptr, const size_t *segm… in SparseSegmentSumKernel()
95 __global__ void SparseSegmentSumKernel(const half *x_ptr, const S *indices_ptr, const size_t *segme… in SparseSegmentSumKernel()
123 __global__ void SparseSegmentSqrtNKernel(const R *x_ptr, const S *indices_ptr, const size_t *segmen… in SparseSegmentSqrtNKernel()
152 __global__ void SparseSegmentSqrtNKernel(const float *x_ptr, const S *indices_ptr, const size_t *se… in SparseSegmentSqrtNKernel()
182 __global__ void SparseSegmentSqrtNKernel(const half *x_ptr, const S *indices_ptr, const size_t *seg… in SparseSegmentSqrtNKernel()
244 …_t CalSparseSegmentCombination(const std::string kernel_type, const R *x_ptr, const S *indices_ptr, in CalSparseSegmentCombination()
|
| D | sparse_segment_grad_impl.cu | 23 __global__ void SparseSegmentPosKernel(const S *indices_ptr, size_t *indices_pos_ptr, size_t idx_se… in SparseSegmentPosKernel()
39 __global__ void SparseSegmentSumGradKernel(const R *grad_ptr, const S *indices_ptr, const S *segmen… in SparseSegmentSumGradKernel()
65 __global__ void SparseSegmentSumGradKernel(const half *grad_ptr, const S *indices_ptr, const S *seg… in SparseSegmentSumGradKernel()
91 __global__ void SparseSegmentSqrtNGradKernel(const R *grad_ptr, const S *indices_ptr, const S *segm… in SparseSegmentSqrtNGradKernel()
118 __global__ void SparseSegmentSqrtNGradKernel(const half *grad_ptr, const S *indices_ptr, const S *s… in SparseSegmentSqrtNGradKernel()
177 …parseSegmentGradCombination(const std::string kernel_type, const R *grad_ptr, const S *indices_ptr, in CalSparseSegmentGradCombination()
|
| D | sparse_segment_mean_grad_impl.cu | 23 __global__ void SparseSegmentPosKernel(const S *indices_ptr, size_t *indices_pos_ptr, size_t idx_se… in SparseSegmentPosKernel()
39 __global__ void SparseSegmentMeanGradKernel(const R *grad_ptr, const S *indices_ptr, const S *segme… in SparseSegmentMeanGradKernel()
66 cudaError_t CalSparseSegmentMeanGrad(const R *grad_ptr, const S *indices_ptr, const S *segment_ids_… in CalSparseSegmentMeanGrad()
|
| D | sparse_slice_grad_impl.cu | 25 __global__ void SparseSliceGradKernel(const DataType *x_ptr, const IndexType *indices_ptr, const In… in SparseSliceGradKernel()
52 CUDA_LIB_EXPORT cudaError_t SparseSliceGrad(const DataType *x_ptr, const IndexType *indices_ptr, in SparseSliceGrad()
|
| D | sparse_split_impl.cu | 25 __global__ void SparseSplitKernel(IndexType *split_dim_ptr, IndexType *indices_ptr, DataType *value… in SparseSplitKernel()
55 CUDA_LIB_EXPORT cudaError_t SparseSplit(IndexType *split_dim_ptr, IndexType *indices_ptr, DataType … in SparseSplit()
|
| D | sparse_slice_impl.cu | 25 __global__ void SparseSliceKernel(const IndexType *indices_ptr, const DataType *values_ptr, const I… in SparseSliceKernel()
56 CUDA_LIB_EXPORT cudaError_t SparseSlice(const IndexType *indices_ptr, const DataType *values_ptr, in SparseSlice()
|
| D | sparse_segment_mean_with_num_segments_impl.cu | 61 …__ void SparseSegmentMeanWithNumSegmentsKernel(const DataType *x_ptr, const IndexType *indices_ptr, in SparseSegmentMeanWithNumSegmentsKernel()
128 __global__ void InputValidCheck(IndexType *indices_ptr, IndexType *segment_ids_ptr, IndexType *num_… in InputValidCheck()
148 const DataType *x_ptr, const IndexType *indices_ptr, const IndexType *segment_ids_ptr, in CalSparseSegmentMeanWithNumSegments()
|
| D | sparse_fill_empty_rows.cu | 79 __global__ void CalElementPerRowsKernel(int64_t *dense_shape_ptr, int64_t *indices_ptr, int64_t *el… in CalElementPerRowsKernel()
104 __global__ void CopyRowIndiceKernel(int64_t *indices_ptr, int64_t *row_indices, int64_t *origin_ind… in CopyRowIndiceKernel()
207 CUDA_LIB_EXPORT cudaError_t SparseFillEmptyRows(int64_t *indices_ptr, Complex<S> *values_ptr, Compl… in SparseFillEmptyRows()
239 CUDA_LIB_EXPORT cudaError_t SparseFillEmptyRows(int64_t *indices_ptr, S *values_ptr, S *default_val… in SparseFillEmptyRows()
|
| D | sparse_segment_mean_impl.cu | 62 __global__ void SparseSegmentMeanKernel(const DataType *x_ptr, const IndexType *indices_ptr, in SparseSegmentMeanKernel()
97 cudaError_t SparseSegmentMean(const DataType *x_ptr, const IndexType *indices_ptr, const IndexType … in SparseSegmentMean()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/ |
| D | gather_int8_coder.cc | 69 auto indices_ptr = reinterpret_cast<int32_t *>(indices_tensor->data()); in DoCode() local
78 auto indices_ptr = reinterpret_cast<int64_t *>(indices_tensor->data()); in DoCode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | gather_nd.c | 53 int32_t *indices_ptr = (int32_t *)indices_tensor->data_; in GatherNdInitOffset() local
61 int64_t *indices_ptr = (int64_t *)indices_tensor->data_; in GatherNdInitOffset() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | gather_int8.cc | 89 auto indices_ptr = reinterpret_cast<int32_t *>(indices_tensor->data()); in DoGather() local
98 auto indices_ptr = reinterpret_cast<int64_t *>(indices_tensor->data()); in DoGather() local
|
| D | gatherNd_int8.cc | 114 auto indices_ptr = reinterpret_cast<int8_t *>(indices_tensor->data()); in InitOffset() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | sparse_fill_empty_rows_fp32.cc | 90 auto indices_ptr = reinterpret_cast<int32_t *>(in_tensors_[kInput_indices]->data()); in RunInferOutputShape() local
140 auto indices_ptr = reinterpret_cast<int32_t *>(in_tensors_[kInput_indices]->data()); in RunOutputData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | one_hot_fp16.c | 35 const int *indices_ptr = indices + i * inner_size; in OneHotToFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | one_hot_fp32.c | 36 const int32_t *indices_ptr = indices + i * inner_size; in OneHotToFp32() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | scatter_nd_binary.cc | 91 auto indices_ptr = indices->data(); in ReSize() local
|
| D | scatter_nd_base.cc | 119 int *indices_ptr = reinterpret_cast<int *>(indices->data()); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | dense_to_csr_sparse_matrix.cc | 91 auto indices_ptr = reinterpret_cast<indiceT *>(ctx.Input(1)->GetData()); in ComputeKernel() local
|
| D | sparse_segment_sum_with_num_segments.cc | 108 auto indices_ptr = reinterpret_cast<indicesT *>(ctx.Input(1)->GetData()); in ComputeKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | sparse_segment_mean_gpu_kernel.cc | 68 auto indices_ptr = GetDeviceAddress<IndexType>(inputs, kIndex1); in LaunchKernel() local
|
| D | sparse_segment_mean_with_num_segments_gpu_kernel.cc | 71 auto indices_ptr = GetDeviceAddress<IndexType>(inputs, kIndex1); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | inplace_op_gpu_kernel.cc | 105 auto indices_ptr = GetDeviceAddress<int64_t>(workspace, kIndex0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_slice_gpu_kernel.cc | 65 auto indices_ptr = GetDeviceAddress<IndexType>(inputs, kIndex0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse_grad/ |
| D | sparse_slice_grad_gpu_kernel.cc | 91 auto indices_ptr = GetDeviceAddress<IndexType>(inputs, kIndex1); in LaunchKernel() local
|