Searched defs:batch_pointers (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | csr_sparse_matrix_to_sparse_tensor_gpu_kernel.cu | 53 __global__ void StackIndices3D(const S *batch_pointers, const S *row_indices, const S *col_indices,… in StackIndices3D()
76 cudaError_t CallStackIndices3D(const S *batch_pointers, const S *row_indices, const S *col_indices,… in CallStackIndices3D()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_sparse_matrix_nnz_op.py | 29 def construct(self, dense_shape, batch_pointers, row_pointers, col_indices, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_matrix_nnz_gpu_kernel.cc | 70 T *batch_pointers = GetDeviceAddress<T>(inputs, kIndex1); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_matrix_transpose_cpu_kernel.cc | 73 #define BATCH_CHECK(batch, batch_pointers, kernel_name_) … argument
181 size_t batch_pointers = x_batch_pointers_size_; in LaunchKernel() local
267 size_t batch_pointers = x_batch_pointers_size_; in LaunchcomplexKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_matrix_transpose.cc | 171 auto batch_pointers = ctx.Input(1)->NumElements(); in SparseMatrixTransposeCompute() local
259 auto batch_pointers = ctx.Input(1)->NumElements(); in SparseMatrixTransposeComputecomplex() local
|