Searched refs:RowMajor2Col8Major (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | matmul_fp32_wrapper.c | 58 RowMajor2Col8Major(src, dst, params_->col_, params_->deep_); in InitMatrixB()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_base.cc | 199 RowMajor2Col8Major(src_data, dst, params_->col_, params_->deep_); in InitMatrixB()
317 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col8Major : RowMajor2Row8Major; in init_global_variable()
322 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col8Major : RowMajor2Row8Major; in init_global_variable()
|
| D | convolution_fp32.cc | 196 …RowMajor2Col8Major(reinterpret_cast<float *>(origin_weight), reinterpret_cast<float *>(packed_weig… in PackWeight()
|
| D | convolution_1x1_fp32.cc | 293 …RowMajor2Col8Major(reinterpret_cast<float *>(origin_weight), reinterpret_cast<float *>(packed_weig… in PackWeight()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_tests.cc | 47 RowMajor2Col8Major(in, out, 10, 9); in TEST_F()
67 RowMajor2Col8Major(in, out, 18, 5); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | matmul_fp32.h | 52 void RowMajor2Col8Major(const float *src_ptr, float *dst_ptr, int row, int col);
|
| D | conv_common_fp32.c | 51 Row2ColMajor = RowMajor2Col8Major; in ConvFp32()
137 Row2ColMajor = RowMajor2Col8Major; in ConvFp32OutNC4HW4()
|
| D | lstm_fp32.c | 34 RowMajor2Col8Major(src_batch, dst_batch, col, deep); in PackLstmWeight()
|
| D | attention_fp32.c | 170 RowMajor2Col8Major(cur_src, cur_dst, deep, real_col); in PackRightMatrix()
|
| D | matmul_fp32.c | 534 void RowMajor2Col8Major(const float *src_ptr, float *dst_ptr, int row, int col) { in RowMajor2Col8Major() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 131 RowMajor2Col8Major(src, dst, param_.col_, param_.deep_); in InitMatrixB()
|