| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | deconvolution_fp32_wrapper.c | 40 …MatMulOpt(packed_input, packed_weight + task_id * thread_stride * C8NUM * kernel_plane * matmul_pa… in DoDeconvFp32()
45 …MatMulOpt(packed_input, packed_weight + task_id * thread_stride * C8NUM * kernel_plane * matmul_pa… in DoDeconvFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | attention_fp32.c | 336 …MatMulOpt(cur_q, cur_wq, cur_q2wq, bq_mat->packed_data_, ActType_No, q_mat->col_, q_mat->row_, wq_… in QWithPosition()
400 …MatMulOpt(cur_k, cur_wk, cur_k2wk, bk_mat->packed_data_, ActType_No, k_mat->col_, k_mat->row_, wk_… in KMulWeightK()
431 …MatMulOpt(cur_v, cur_wv, cur_v2wv, bv_mat->packed_data_, ActType_No, v_mat->col_, v_mat->row_, wv_… in VMulWeightV()
463 …MatMulOpt(cur_p, cur_wp, cur_p2wp, NULL, ActType_No, p_mat->col_, p_mat->row_, wp_mat->col_, wp_ma… in PMulWeightP()
497 …MatMulOpt(q2wq_with_pu_trans_mat->packed_data_, k2wk_trans_mat->packed_data_, logits_with_u_mat->d… in CalculateLogits()
502 …MatMulOpt(q2wq_with_pv_trans_mat->packed_data_, p2wp_trans_mat->packed_data_, logits_with_v_mat->d… in CalculateLogits()
538 …MatMulOpt(cur_logits, cur_v2wv, cur_logits2v, NULL, ActType_No, softmax_mat->col_, softmax_mat->ro… in RelPosAttention()
565 …MatMulOpt(cur_concat_out, cur_wo, cur_output, bo_mat->packed_data_, ActType_No, logits2v_trans_mat… in RelPosAttention()
|
| D | gru_fp32.c | 26 MatMulOpt(a, b, c, bias, ActType_No, deep, row, col, col, OutType_Nhwc); in GruMatMul()
102 MatMulOpt(packed_input, weight_loop, gate_loop, bias_loop, ActType_No, gru_param->input_size_, in GruUnidirectional()
|
| D | conv_winograd_fp32.c | 83 …MatMulOpt(tmp_col_ptr, trans_weight + i * in_channel * oc_tile * col_tile, dst_ptr + i * C8NUM, NU… in ConvWinogardFp32()
|
| D | lstm_fp32.c | 83 MatMulOpt(a, b, c, bias, ActType_No, deep, row, col, col, OutType_Nhwc); in LstmMatMul()
240 MatMulOpt(packed_input, weight_loop, gate_loop, bias_loop, ActType_No, lstm_param->input_size_, in LstmUnidirectional()
|
| D | matmul_fp32.h | 39 void MatMulOpt(const float *a, const float *b, float *c, const float *bias, ActType act_type, int d…
|
| D | matmul_fp32.c | 1046 void MatMulOpt(const float *a, const float *b, float *c, const float *bias, ActType act_type, int d… in MatMulOpt() function
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_1x1_fp32.cc | 154 …MatMulOpt(pack_input_, reinterpret_cast<float *>(packed_weight_) + task_id * thread_stride_ * matm… in DoConv1x1()
158 …MatMulOpt(pack_input_, reinterpret_cast<float *>(packed_weight_) + task_id * thread_stride_ * matm… in DoConv1x1()
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()
|
| D | deconvolution_fp32.cc | 151 MatMulOpt( in DoDeconv()
|
| D | matmul_fp32_base.cc | 298 …MatMulOpt(batch_a_ptr_, b, c, bias, params_->act_type_, params_->deep_, params_->row_, cur_oc, par… in FloatRun()
|
| D | lstm_fp32.cc | 338 …MatMulOpt(input, b, c, bias, ActType_No, lstm_param_->input_size_, lstm_param_->seq_len_ * lstm_pa… in InputWeightMatMul()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | gemm.c | 598 MatMulOpt(mat_a_input, mat_b_input, output, gcb->bias, gcb->atype, K, M, N, ldc, OutType_Nhwc);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 212 …MatMulOpt(batch_a_ptr_, b, output, bias, param_.act_type_, param_.deep_, param_.row_, SizeToInt(cu… in FloatRun()
|