Searched defs:MatrixTranspose (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | matrix_transpose_impl.cu | 39 cudaError_t MatrixTranspose(const T *input, int elements, int row, int col, T *output, uint32_t dev… in MatrixTranspose() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | minimal_filtering_generator.c | 158 void MatrixTranspose(const float *matrix, float *trans_matrix, int row, int col) { in MatrixTranspose() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_matrix_sparse_matmul_gpu_kernel.cc | 23 void SparseMatrixSparseMatMulGpuKernelMod::MatrixTranspose(int m, int n, int nnz, void *csrVal, con… in MatrixTranspose() function in mindspore::kernel::SparseMatrixSparseMatMulGpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/ |
| D | common_utils.cc | 719 NodePtr MatrixTranspose(BpropBuilder *ib, const NodePtr &x) { in MatrixTranspose() function
|