Home
last modified time | relevance | path

Searched defs:vector_sum_col (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuGemmLowpOffsetContributionKernel.cpp47 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 DCpuGemmLowpOffsetContributionOutputStageKernel.cpp284 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 DCpuGemmLowpMatrixReductionKernel.cpp355 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 DClGemmLowpOffsetContributionKernel.cpp45 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 DClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp49const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate_arguments()
182ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in validate_and_configure_window()
291ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in configure()
398const 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 DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp58const ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, const ITensorInfo *bias, in validate_arguments()
198ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in validate_and_configure_window()
290ITensorInfo *vector_sum_col, const ITensorInfo *vector_sum_row, ITensorInfo *bias, in configure()
436const 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 DClGemmLowpOffsetContributionOutputStageKernel.cpp47 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 DClGemmLowpReductionKernel.cpp155 …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 DCpuGemmLowpMatrixMultiplyCore.cpp517 … 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 Dcl_gemmlowp_reshaped_rhs_only_fused_output_stage_fixedpoint.cpp362 CLTensor vector_sum_col{}; member in CLGEMMLowpMatrixMultiplyReshapedOnlyRHSFusedOutputStageFixedpointExample