Searched refs:gemm_input_to_use (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClGemmConv2d.cpp | 194 const ITensorInfo *gemm_input_to_use = src; in configure() local
246 gemm_input_to_use = &_im2col_output; in configure()
321 …configure_mm(compile_context, gemm_input_to_use, &_weights_reshaped, biases_to_use, gemm_output_to… in configure()
380 const ITensorInfo *gemm_input_to_use = src; in validate() local
457 gemm_input_to_use = &im2col_reshaped_info; in validate()
465 shape_gemm = gemm_input_to_use->tensor_shape(); in validate()
523 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemm_input_to_use, weights_to_use, biases_to_use, gemm_out… in validate()
549 auto gemm_input_to_use = src; in run() local
565 gemm_input_to_use = im2col_output.get(); in run()
572 pack_mm.add_const_tensor(TensorType::ACL_SRC_0, gemm_input_to_use); in run()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuGemmConv2d.cpp | 287 const ITensorInfo *gemm_input_to_use = src; in configure() local
329 gemm_input_to_use = &_im2col_output; in configure()
364 …configure_mm(gemm_input_to_use, &_weights_reshaped, biases, gemm_output_to_use, act_info, enable_f… in configure()
452 const ITensorInfo *gemm_input_to_use = src; in validate() local
517 gemm_input_to_use = &im2col_reshaped_info; in validate()
524 TensorShape shape_gemm = gemm_input_to_use->tensor_shape(); in validate()
537 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemm_input_to_use, weights_to_use, biases, gemm_output_to_… in validate()
555 auto gemm_input_to_use = src; in run() local
572 gemm_input_to_use = im2col_output.get(); in run()
594 pack_mm.add_const_tensor(TensorType::ACL_SRC_0, gemm_input_to_use); in run()
|