| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | matrix_triangular_solve_gpu_kernel.cc | 38 int ldb) { in Trsm()
43 double *B, int ldb) { in Trsm()
48 Complex<float> *B, int ldb) { in Trsm()
54 Complex<double> *B, int ldb) { in Trsm()
61 … const float *const A[], int lda, float *B[], int ldb, int batch_size) { in TrsmBatched()
66 … const double *const A[], int lda, double *B[], int ldb, int batch_size) { in TrsmBatched()
71 … Complex<float> *A[], int lda, Complex<float> *B[], int ldb, int batch_size) { in TrsmBatched()
77 … Complex<double> *A[], int lda, Complex<double> *B[], int ldb, int batch_size) { in TrsmBatched()
|
| D | cholesky_solve_gpu_kernel.h | 43 const float *A, int lda, float *B, int ldb) { in cublasXtrsm()
48 const double *A, int lda, double *B, int ldb) { in cublasXtrsm()
53 … const float *alpha, const float *const A[], int lda, float *const B[], int ldb, in cublasXtrsmBatched()
60 int ldb, int batchCount) { in cublasXtrsmBatched()
|
| D | matmul_gpu_kernel.cc | 157 const int ldb = (transpose_x2_ != CUBLAS_OP_N) ? SizeToInt(k_) : SizeToInt(n_); in LaunchKernel() local
|
| D | update_thor_gradient.h | 85 const int ldb = SizeToInt(gradient_size.ori_w + gradient_size.pad_w); in Launch() local
|
| D | einsum_helper.h | 215 const int ldb = (transpose_x2_ == CUBLAS_OP_T) ? SizeToInt(k) : SizeToInt(n); in Bmm() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | triangle_matrix_copy_impl.cu | 21 const size_t count, const size_t ldb, const size_t m) { in TriangleMatrixCopyKernel()
55 const size_t ldb, const size_t m, cudaStream_t cuda_stream) { in TriangleMatrixCopy()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/cuda_impl/ |
| D | cublas_utils.cc | 37 const int ldb = (trans_b == CUBLAS_OP_N) ? n : k; in CublasMM1Batch() local
59 const int ldb = (trans_b == CUBLAS_OP_N) ? n : k; in CublasMMBatched() local
81 const int ldb = lds[1]; in CublasGemmWrapper() local
104 const int ldb = lds[1]; in CublasGemmStridedBatchedWrapper() local
134 const int ldb = lds[1]; in CublasLtGemmWrapper() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/ |
| D | gemm_fp16.c | 339 …const float16_t *mat_b, int ldb, float16_t beta, float16_t *mat_c, int ldc, float16_t *workspace) { in GemmMatmulFp16()
349 const float16_t *mat_b, int ldb, float16_t beta, float16_t *mat_c, int ldc, in GemmMatmulPlusFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | gemm.c | 777 int ldb, float beta, float *mat_c, int ldc, float *workspace) {
789 int ldb, float beta, float *mat_c, int ldc, float *workspace, GemmCb *gcb) {
|