Searched defs:matrix_row (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | tril_triu_impl.cu | 25 …global__ void Tril(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in Tril()
37 …global__ void Triu(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in Triu()
49 … Triu(const size_t size, const Complex<float> *input, const int diagonal, const int64_t matrix_row, in Triu()
64 …Triu(const size_t size, const Complex<double> *input, const int diagonal, const int64_t matrix_row, in Triu()
79 cudaError_t CalTril(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in CalTril()
87 cudaError_t CalTriu(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in CalTriu()
|
| D | ger_impl.cu | 20 …obal__ void Ger(const size_t size, const T *row_input, const T *col_input, const size_t matrix_row, in Ger()
32 …aError_t CalGer(const size_t size, const T *row_input, const T *col_input, const size_t matrix_row, in CalGer()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | cholesky_inverse_gpu_kernel.cc | 66 int64_t matrix_row = input_shape[kZero]; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | matrix_triangular_solve.cc | 82 int64_t matrix_row = matrix_shape[matrix_shape.size() - kIndex2]; in MatrixTriangularSolveInferShape() local
|
| D | matrix_solve_ls.cc | 66 int64_t matrix_row = matrix_last[0]; in MatrixSolveLsInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | matmul_fp32_avx512.cc | 367 int matrix_row = params_->row_; in ParallelRunByGEMM() local
443 int matrix_row = params_->row_; in ParallelRunByBatchColRowGEMM() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | matmul_avx512.c | 486 int matrix_row = matmul->compute_.row_; in MatmulAVX512ParallelRunByGEMM() local
645 int matrix_row = matmul->compute_.row_; in MatmulAVX512ParallelRunByBatchColRowGEMM() local
|