Searched refs:gemm_output_to_use (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuGemmConv2d.cpp | 288 ITensorInfo *gemm_output_to_use = dst; in configure() local
348 gemm_output_to_use = &_gemm_output; in configure()
357 gemm_output_to_use = &_gemm_output_3d; in configure()
364 …configure_mm(gemm_input_to_use, &_weights_reshaped, biases, gemm_output_to_use, act_info, enable_f… in configure()
370 _col2im_kernel->configure(gemm_output_to_use, dst, Size2D(conv_w, conv_h)); in configure()
376 _reshape_kernel->configure(gemm_output_to_use, dst); in configure()
453 const ITensorInfo *gemm_output_to_use = dst; in validate() local
534 gemm_output_to_use = &info_gemm; in validate()
537 …RN_ON_ERROR(validate_mm(gemm_input_to_use, weights_to_use, biases, gemm_output_to_use, act_info, e… in validate()
543 …ARM_COMPUTE_RETURN_ON_ERROR(kernels::CpuCol2ImKernel::validate(gemm_output_to_use, dst, Size2D(con… in validate()
[all …]
|
| H A D | CpuGemm.cpp | 100 ITensorInfo *gemm_output_to_use = (_run_bias_addition) ? &_tmp_d : d; in configure() local
108 _mm_kernel->configure(a, b, gemm_output_to_use, alpha, false); in configure()
127 …_mm_kernel->configure(&_tmp_a, &_tmp_b, gemm_output_to_use, alpha, true, GEMMReshapeInfo(m, n, k)); in configure()
133 _add_bias->configure(gemm_output_to_use, c, d, ConvertPolicy::SATURATE); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClGemmConv2d.cpp | 195 ITensorInfo *gemm_output_to_use = dst; in configure() local
263 gemm_output_to_use = &_gemm_output; in configure()
321 …ile_context, gemm_input_to_use, &_weights_reshaped, biases_to_use, gemm_output_to_use, gemmlowp_ou… in configure()
330 …_col2im_kernel->configure(compile_context, gemm_output_to_use, dst, Size2D(conv_w, conv_h), conv2d… in configure()
381 const ITensorInfo *gemm_output_to_use = dst; in validate() local
471 gemm_output_to_use = &info_gemm; in validate()
523 …RROR(validate_mm(gemm_input_to_use, weights_to_use, biases_to_use, gemm_output_to_use, gemmlowp_ou… in validate()
529 …ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClCol2ImKernel::validate(gemm_output_to_use, dst, Size2D(conv… in validate()
550 auto gemm_output_to_use = dst; in run() local
569 gemm_output_to_use = gemm_output.get(); in run()
[all …]
|