Searched defs:matrix_col (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | tril_triu_impl.cu | 26 const int64_t matrix_col, T *output) { in Tril()
38 const int64_t matrix_col, T *output) { in Triu()
50 const int64_t matrix_col, Complex<float> *output) { in Triu()
65 const int64_t matrix_col, Complex<double> *output) { in Triu()
80 … const int64_t matrix_col, T *output, const uint32_t &device_id, cudaStream_t cuda_stream) { in CalTril()
88 … const int64_t matrix_col, T *output, const uint32_t &device_id, cudaStream_t cuda_stream) { in CalTriu()
|
| D | ger_impl.cu | 21 const size_t matrix_col, T *output) { in Ger()
33 … const size_t matrix_col, T *output, const uint32_t &device_id, cudaStream_t cuda_stream) { in CalGer()
|
| D | trace_impl.cu | 40 __global__ void Trace(const T *input, const int64_t sum_size, const int64_t matrix_col, T *output) { in Trace()
74 cudaError_t CalTrace(const T *input, const int64_t sum_size, const int64_t matrix_col, T *output, in CalTrace()
|
| D | trace_grad_impl.cu | 23 S matrix_col = input_shape[1]; in TraceGrad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | matmul_fp32_avx512.cc | 270 int matrix_col = col_step_; in ParallelRunByRow1Deep1GEPDOT() local
316 int matrix_col = col_step_; in ParallelRunByGEPM() local
368 int matrix_col = col_step_; in ParallelRunByGEMM() local
444 int matrix_col = col_step_; in ParallelRunByBatchColRowGEMM() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | cholesky_inverse_gpu_kernel.cc | 67 int64_t matrix_col = input_shape[kOne]; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | matmul_avx512.c | 440 int matrix_col = matmul->compute_.col_step_; in MatmulAVX512ParallelRunByGEPM() local
487 int matrix_col = matmul->compute_.col_step_; in MatmulAVX512ParallelRunByGEMM() local
599 int matrix_col = matmul->compute_.col_step_; in MatmulAVX512ParallelRunByRow1Deep1GEPDOT() local
646 int matrix_col = matmul->compute_.col_step_; in MatmulAVX512ParallelRunByBatchColRowGEMM() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | matrix_triangular_solve.cc | 83 int64_t matrix_col = matrix_shape[matrix_shape.size() - kIndex1]; in MatrixTriangularSolveInferShape() local
|
| D | matrix_solve_ls.cc | 67 int64_t matrix_col = matrix_last[1]; in MatrixSolveLsInferShape() local
|