Searched refs:thread_pack_input (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_1x1_fp32.cc | 183 float *thread_pack_input = pack_input_ + task_id * row_tile_ * matmul_param_->deep_; in DoConv1x1Hw() local
194 PackMatmulInput(cur_intput, thread_pack_input, cur_rows, matmul_param_->deep_); in DoConv1x1Hw()
196 MatMulOpt(thread_pack_input, reinterpret_cast<float *>(packed_weight_), cur_output, in DoConv1x1Hw()
201 MatMulOpt(thread_pack_input, reinterpret_cast<float *>(packed_weight_), cur_output, in DoConv1x1Hw()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_1x1_fp16.cc | 240 float16_t *thread_pack_input = pack_input_ + task_id * thread_stride_ * matmul_param_->deep_; in RunHw() local
243 RowMajor2Col12MajorFp16Opt(thread_input_ptr, thread_pack_input, cur_hw_, matmul_param_->deep_); in RunHw()
245 …MatMul12x16Fp16Opt(thread_pack_input, reinterpret_cast<float16_t *>(packed_weight_), thread_output… in RunHw()
249 …MatMul12x8A32Fp16(thread_pack_input, reinterpret_cast<float16_t *>(packed_weight_), thread_output_… in RunHw()
|