24 #ifndef ARM_COMPUTE_CL_GEMM_HELPERS_H 25 #define ARM_COMPUTE_CL_GEMM_HELPERS_H 55 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
configure_lhs_rhs_info(
unsigned int m,
unsigned int n,
unsigned int m0,
unsigned int n0,
unsigned int k0,
unsigned int v0,
unsigned int h0,
56 bool lhs_interleave,
bool rhs_interleave,
bool lhs_transpose,
bool rhs_transpose,
bool export_to_cl_image =
false);
73 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
select_lhs_rhs_info(std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> info_img,
74 std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo> info_buf,
void update_padding_for_cl_image(ITensorInfo *tensor)
Update padding required to export the OpenCL buffer to OpenCL image2d.
Copyright (c) 2017-2021 Arm Limited.
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > select_lhs_rhs_info(std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > info_img, std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > info_buf, unsigned int n, unsigned int k, unsigned int b, DataType data_type)
Select GEMMLHSMatrixInfo and GEMMRHSMatrixInfo.
Status validate_image2d_support_on_rhs(const ITensorInfo &tensor_reshaped_info, const GEMMRHSMatrixInfo &rhs_info)
Utility function to validate the image2d OpenCL object support on the RHS reshaped matrix...
std::pair< GEMMLHSMatrixInfo, GEMMRHSMatrixInfo > configure_lhs_rhs_info(unsigned int m, unsigned int n, unsigned int m0, unsigned int n0, unsigned int k0, unsigned int v0, unsigned int h0, bool lhs_interleave, bool rhs_interleave, bool lhs_transpose, bool rhs_transpose, bool export_to_cl_image)
Configure GEMMLHSMatrixInfo and GEMMRHSMatrixInfo.
UniqueGemmCommon< Top, Tret > gemm(const GemmArgs &args, const OutputStage &={})
DataType
Available data types.