21.08
|
OpenCL GEMM kernel selection parameters. More...
#include <CLTypes.h>
Data Fields | |
unsigned int | m { 0 } |
Number of rows for the lhs matrix. More... | |
unsigned int | n { 0 } |
Number of columns for the rhs matrix. More... | |
unsigned int | k { 0 } |
Number of rows for the rhs matrix. More... | |
unsigned int | b { 0 } |
Batch size. More... | |
bool | is_rhs_constant { false } |
True if the content of the rhs matrix is constant. More... | |
DataType | data_type { DataType::UNKNOWN } |
Data type. More... | |
OpenCL GEMM kernel selection parameters.
These information are retrieved to select the GEMM kernel on OpenCL
unsigned int b { 0 } |
Batch size.
Definition at line 57 of file CLTypes.h.
Referenced by CLGEMMDefaultTypeBifrost::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeMidgard::select_kernel().
DataType data_type { DataType::UNKNOWN } |
Data type.
Definition at line 59 of file CLTypes.h.
Referenced by CLGEMMDefaultTypeBifrost::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeMidgard::select_kernel().
bool is_rhs_constant { false } |
True if the content of the rhs matrix is constant.
Definition at line 58 of file CLTypes.h.
Referenced by CLGEMMDefaultTypeBifrost::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeMidgard::select_kernel().
unsigned int k { 0 } |
Number of rows for the rhs matrix.
Rhs matrix NOT transposed
Definition at line 56 of file CLTypes.h.
Referenced by CLGEMMDefaultTypeBifrost::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeMidgard::select_kernel().
unsigned int m { 0 } |
Number of rows for the lhs matrix.
Lhs matrix NOT transposed
Definition at line 54 of file CLTypes.h.
Referenced by arm_compute::cl_gemm::auto_heuristics::select_default_gemm_kernel(), CLGEMMDefaultTypeMidgard::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeBifrost::select_kernel().
unsigned int n { 0 } |
Number of columns for the rhs matrix.
Rhs matrix NOT transposed
Definition at line 55 of file CLTypes.h.
Referenced by CLGEMMDefaultTypeBifrost::select_kernel(), CLGEMMDefaultTypeValhall::select_kernel(), and CLGEMMDefaultTypeMidgard::select_kernel().