| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | attention_fp32.c | 337 wq_mat->col_, OutType_Nhwc); in QWithPosition()
401 wk_mat->col_, OutType_Nhwc); in KMulWeightK()
432 wv_mat->col_, OutType_Nhwc); in VMulWeightV()
464 OutType_Nhwc); in PMulWeightP()
499 logits_with_u_mat->col_, OutType_Nhwc); in CalculateLogits()
504 logits_with_v_mat->col_, OutType_Nhwc); in CalculateLogits()
539 v2wv_trans_mat->col_, v2wv_trans_mat->col_, OutType_Nhwc); in RelPosAttention()
566 logits2v_trans_mat->row_, wo_mat->col_, wo_mat->col_, OutType_Nhwc); in RelPosAttention()
|
| D | gru_fp32.c | 26 MatMulOpt(a, b, c, bias, ActType_No, deep, row, col, col, OutType_Nhwc); in GruMatMul()
103 …ram->seq_len_ * gru_param->batch_, gru_param->hidden_size_, gru_param->hidden_size_, OutType_Nhwc); in GruUnidirectional()
|
| D | conv_common_fp32.c | 92 real_cal_row, out_channel, (size_t)out_channel, (size_t)OutType_Nhwc); in ConvFp32()
95 real_cal_row, out_channel, (size_t)out_channel, (int)OutType_Nhwc); in ConvFp32()
98 real_cal_row, out_channel, out_channel, OutType_Nhwc); in ConvFp32()
101 out_channel, out_channel, OutType_Nhwc); in ConvFp32()
104 out_channel, out_channel, OutType_Nhwc); in ConvFp32()
|
| D | lstm_fp32.c | 83 MatMulOpt(a, b, c, bias, ActType_No, deep, row, col, col, OutType_Nhwc); in LstmMatMul()
242 OutType_Nhwc); in LstmUnidirectional()
|
| D | matmul_fp32.c | 986 if (out_type == OutType_Nhwc) { in MatMul12x8()
1057 } else if (out_type == OutType_Nhwc) { in MatMulOpt()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_1x1_fp16.cc | 217 matmul_param_->row_, cur_oc, matmul_param_->col_, OutType_Nhwc); in RunOc()
222 matmul_param_->row_, cur_oc, matmul_param_->col_, OutType_Nhwc); in RunOc()
247 cur_hw_, matmul_param_->col_, matmul_param_->col_, OutType_Nhwc); in RunHw()
251 cur_hw_, matmul_param_->col_, matmul_param_->col_, OutType_Nhwc); in RunHw()
|
| D | matmul_base_fp16.cc | 305 params_->col_, OutType_Nhwc); in RunImpl()
308 OutType_Nhwc); in RunImpl()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_fp16.c | 60 … real_cal_row, conv_param->output_channel_, conv_param->output_channel_, OutType_Nhwc); in ConvFp16()
106 real_in_row, real_weight_row, real_weight_row, OutType_Nhwc); in ConvOutNc8hw8Fp16()
148 real_weight_row, real_weight_row, OutType_Nhwc); in Conv1x1OutNc8hw8MultiThreadByInputFp16()
181 real_weight_row, real_weight_row, OutType_Nhwc); in Conv1x1OutNc8hw8MultiThreadByWeightFp16()
|
| D | lstm_fp16.c | 177 MatMulFp16(a, b, c, bias, ActType_No, deep, row, col, col, OutType_Nhwc); in LstmMatMulFp16()
256 OutType_Nhwc); in LstmUnidirectionalFp16()
|
| D | gru_fp16.c | 99 …ram->seq_len_ * gru_param->batch_, gru_param->hidden_size_, gru_param->hidden_size_, OutType_Nhwc); in GruUnidirectionalFp16()
|
| D | matmul_fp16.c | 231 if (write_mode == OutType_Nhwc) { in MatMul16x8Fp16()
293 if (write_mode == OutType_Nhwc) { // common conv and matmul in MatMul12x8Fp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | matmul_parameter.h | 36 typedef enum OutType { OutType_C8 = 0, OutType_Nhwc = 1, OutType_TileC8 = 2, OutType_NC4HW4 = 3 } O… enumerator
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_1x1_fp32.cc | 156 matmul_param_->row_, cur_oc, matmul_param_->col_, OutType_Nhwc); in DoConv1x1()
198 matmul_param_->col_, matmul_param_->col_, OutType_Nhwc); in DoConv1x1Hw()
|
| D | matmul_fp32_base.cc | 299 OutType_Nhwc); in FloatRun()
|
| D | lstm_fp32.cc | 339 cur_oc, lstm_param_->hidden_size_, OutType_Nhwc); in InputWeightMatMul()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | gemm_fp16.c | 381 MatMulFp16(mat_a_input, mat_b_input, output, gcb->bias, gcb->atype, K, M, N, ldc, OutType_Nhwc); in GemmMatmulPlusFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
| D | MatMul_Sse.c | 203 if (write_mode == OutType_Nhwc) c = dst - col; in MatmulFloatSse64Opt()
|
| /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 | 213 param_.col_, OutType_Nhwc); in FloatRun()
|