24 #ifndef ARM_COMPUTE_ICL_GEMM_KERNEL_CONFIG_H 25 #define ARM_COMPUTE_ICL_GEMM_KERNEL_CONFIG_H 60 : _configs{ func_f32, func_f16, func_int8 }
75 return _configs.at(DT_F32);
77 return _configs.at(DT_F16);
81 return _configs.at(DT_INT8);
88 std::array<T, 3> _configs;
114 virtual std::pair<GEMMLHSMatrixInfo, GEMMRHSMatrixInfo>
configure(
unsigned int m,
unsigned int n,
unsigned int k,
unsigned int b,
DataType data_type) = 0;
static constexpr size_t DT_F32
Alias for F32 index.
Basic container for the OpenCL GEMM configuration functions.
Basic interface for the GEMM kernel configuration.
1 channel, 1 F32 per channel
#define ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(TypeName)
CLGEMMConfigArray(T func_f32, T func_f16, T func_int8)
Constructor.
static constexpr size_t DT_INT8
Alias for Int8 index.
Copyright (c) 2017-2021 Arm Limited.
1 channel, 1 F16 per channel
quantized, asymmetric fixed-point 8-bit number unsigned
quantized, symmetric per channel fixed-point 8-bit number
T get_function(DataType data_type)
Method to return the GEMM configuration function based on data type.
IClGemmKernelConfig(GPUTarget arch)
Constructor.
GPUTarget
Available GPU Targets.
static constexpr size_t DT_F16
Alias for F16 index.
UniqueGemmCommon< Top, Tret > gemm(const GemmArgs &args, const OutputStage &={})
quantized, asymmetric fixed-point 8-bit number signed
im2col_func configure(src_target.info(), dst_target.info(), spatial_kernel, conv_info, has_bias)
DataType
Available data types.