/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
H A D | CpuGemmLowpOffsetContributionKernel.cpp | 47 Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const IT… in validate_arguments() 98 … ITensor *mm_result, const ITensor *vector_sum_col, const ITensor *vector_sum_row, in run_offset_contribution() 365 …owpOffsetContributionKernel::configure(ITensorInfo *mm_result, ITensorInfo *vector_sum_col, ITenso… in configure() 390 …tContributionKernel::validate(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, con… in validate() 403 auto vector_sum_col = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run_op() local
|
H A D | CpuGemmLowpOffsetContributionOutputStageKernel.cpp | 284 inline Iterator get_vector_sum_col_it(const Window &window, const ITensor *vector_sum_col) in get_vector_sum_col_it() 477 …const ITensor *mm_result, const ITensor *vector_sum_col, const ITensor *vector_sum_row, const ITen… in run_offset_contribution_output_stage() 681 …const ITensor *mm_result, const ITensor *vector_sum_col, const ITensor *vector_sum_row, const ITen… in run_offset_contribution_output_stage_symm() 781 Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const IT… in validate_arguments() 855 …butionOutputStageKernel::configure(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, in configure() 891 …ibutionOutputStageKernel::validate(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, in validate() 907 auto vector_sum_col = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run_op() local
|
H A D | CpuGemmLowpMatrixReductionKernel.cpp | 355 auto vector_sum_col = reinterpret_cast<int32_t *>(out.ptr()); in run_internal() local
|
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
H A D | ClGemmLowpOffsetContributionKernel.cpp | 45 Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const IT… in validate_arguments() 109 …const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row… in configure() 166 …tContributionKernel::validate(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, con… in validate() 178 …const auto vector_sum_col = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const… in run_op() local
|
H A D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 49 … const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate_arguments() 182 … ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in validate_and_configure_window() 291 … ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in configure() 398 … const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate() 426 …const auto vector_sum_col = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const… in run_op() local
|
H A D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp | 58 … const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate_arguments() 198 … ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in validate_and_configure_window() 290 … ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in configure() 436 … const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate() 464 …const auto vector_sum_col = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_c… in run_op() local
|
H A D | ClGemmLowpOffsetContributionOutputStageKernel.cpp | 47 Status validate_arguments(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const IT… in validate_arguments() 133 …const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row… in configure() 211 …onOutputStageKernel::validate(const ITensorInfo *mm_result, const ITensorInfo *vector_sum_col, con… in validate() 226 …const auto vector_sum_col = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_c… in run_op() local
|
H A D | ClGemmLowpReductionKernel.cpp | 155 …t CLCompileContext &compile_context, const ITensorInfo *mtx_b, ITensorInfo *vector_sum_col, const … in configure() 192 …pMatrixBReductionKernel::validate(const ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, con… in validate()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
H A D | CpuGemmLowpMatrixMultiplyCore.cpp | 517 … CpuAuxTensorHandler vector_sum_col(offset_int_vec(VectorSumCol), _vector_sum_col, tensors, false); in run() local 698 CpuAuxTensorHandler vector_sum_col(_vector_sum_col, *vector_sum_col_p); in prepare() local
|
/aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/ |
H A D | cl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp | 362 CLTensor vector_sum_col{}; member in CLGEMMLowpMatrixMultiplyReshapedOnlyRHSFusedOutputStageFixedpointExample
|