Home
last modified time | relevance | path

Searched refs:compute_mm_shape (Results 1 – 14 of 14) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/kernels/
DClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp138 …r_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_arguments()
154 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_and_configure_window()
214 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in configure()
DClGemmMatrixMultiplyNativeKernel.cpp132 …r_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_arguments()
162 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_and_configure_window()
233 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in configure()
DClGemmLowpMatrixMultiplyNativeKernel.cpp97 …r_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_arguments()
124 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_and_configure_window()
DClGemmLowpMatrixMultiplyReshapedKernel.cpp94 …const TensorInfo tensor_info_dst = dst->clone()->set_tensor_shape(compute_mm_shape(*src0, *src1, g… in validate_arguments()
111 …auto_init_if_empty(*dst, src0->clone()->set_tensor_shape(compute_mm_shape(*src0, *src1, gemm_info)… in validate_and_configure_window()
DClGemmMatrixMultiplyReshapedKernel.cpp138 …r_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_arguments()
195 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in configure()
DClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp130 …r_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in validate_arguments()
194 …empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(*src0, *src1,… in configure()
DClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp90 const TensorShape expected_dst_shape = compute_mm_shape(*src0, *src1, gemm_info); in validate_arguments()
200 const TensorShape expected_dst_shape = compute_mm_shape(*src0, *src1, gemm_info); in validate_and_configure_window()
DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp105 const TensorShape expected_dst_shape = compute_mm_shape(*src0, *src1, gemm_info); in validate_arguments()
220 const TensorShape expected_dst_shape = compute_mm_shape(*src0, *src1, gemm_info); in validate_and_configure_window()
/external/ComputeLibrary/tests/validation/CL/
DGEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp145 const TensorShape dst_shape = compute_mm_shape(TensorInfo(lhs_shape, 1, DataType::QASYMM8), in validate_configuration()
DGEMMMatrixMultiplyNative.cpp275 const TensorShape dst_shape = compute_mm_shape(TensorInfo(lhs_shape, 1, data_type), in validate_configuration()
DGEMMMatrixMultiplyReshapedOnlyRHS.cpp303 const TensorShape dst_shape = compute_mm_shape(TensorInfo(lhs_shape, 1, dt_input0), in validate_configuration()
/external/ComputeLibrary/src/gpu/cl/operators/
DClGemmLowpMatrixMultiplyCore.cpp93 …auto_init_if_empty(mm_result_s32_info, a->clone()->set_tensor_shape(compute_mm_shape(*a, *b, false… in validate_lhs_rhs_info_native()
589 …auto_init_if_empty(mm_result_s32_info, a->clone()->set_tensor_shape(compute_mm_shape(*matrix_a_inf… in validate()
597 …auto_init_if_empty(mm_result_s32_info, a->clone()->set_tensor_shape(compute_mm_shape(*matrix_a_inf… in validate()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h905 inline TensorShape compute_mm_shape(const ITensorInfo &input0, const ITensorInfo &input1, bool is_i… in compute_mm_shape() function
941 inline TensorShape compute_mm_shape(const ITensorInfo &input0, const ITensorInfo &input1, const GEM… in compute_mm_shape() function
979 inline TensorShape compute_mm_shape(const ITensorInfo &input0, const ITensorInfo &input1, const GEM… in compute_mm_shape() function
/external/ComputeLibrary/src/cpu/operators/
DCpuGemm.cpp258 …_if_empty(tmp_output_info, matrix_a_info->clone()->set_tensor_shape(compute_mm_shape(*matrix_a_inf… in validate()