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()
|