Searched refs:RowMajor2Col16Major (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_fp32.cc | 190 …RowMajor2Col16Major(reinterpret_cast<float *>(origin_weight), reinterpret_cast<float *>(packed_wei… in PackWeight()
|
| D | convolution_1x1_fp32.cc | 287 …RowMajor2Col16Major(reinterpret_cast<float *>(origin_weight), reinterpret_cast<float *>(packed_wei… in PackWeight()
|
| D | matmul_fp32_base.cc | 307 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col16Major : RowMajor2Row16Major; in init_global_variable()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | matmul_fp32.h | 54 void RowMajor2Col16Major(const float *src_ptr, float *dst_ptr, int row, int col);
|
| D | lstm_fp32.c | 30 RowMajor2Col16Major(src_batch, dst_batch, col, deep); in PackLstmWeight()
|
| D | attention_fp32.c | 156 RowMajor2Col16Major(cur_src, cur_dst, deep, real_col); in PackRightMatrix()
|
| D | matmul_fp32.c | 616 void RowMajor2Col16Major(const float *src_ptr, float *dst_ptr, int row, int col) { in RowMajor2Col16Major() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 119 RowMajor2Col16Major(src, dst, param_.col_, param_.deep_); in InitMatrixB()
|