Searched refs:RowMajor2Row6Major (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | matmul_fp32.h | 45 void RowMajor2Row6Major(const float *src_ptr, float *dst_ptr, int row, int col);
|
| D | attention_fp32.c | 95 RowMajor2Row6Major(cur_src, cur_dst, real_row, deep); in PackLeftMatrix()
|
| D | matmul_fp32.c | 55 void RowMajor2Row6Major(const float *src_ptr, float *dst_ptr, int row, int col) { in RowMajor2Row6Major() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 70 RowMajor2Row6Major(src, dst, param_.deep_, param_.row_); in InitMatrixA()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_base.cc | 306 matrix_a_pack_fun_ = params_->a_transpose_ ? RowMajor2Row6Major : RowMajor2Col6Major; in init_global_variable()
|