24.04
|
Go to the documentation of this file.
24 #ifndef SRC_RUNTIME_HEURISTICS_INDIRECT_CONV_ICLINDIRECTCONVKERNELCONFIG
25 #define SRC_RUNTIME_HEURISTICS_INDIRECT_CONV_ICLINDIRECTCONVKERNELCONFIG
35 namespace cl_indirect_conv
68 return _configs.at(
DT_F32);
70 return _configs.at(
DT_F16);
77 std::array<T, 2> _configs;
Basic interface for the indirect convolution kernel configuration.
SimpleTensor< float > src
virtual ~IClIndirectConvKernelConfig()=default
Virtual destructor.
ClIndirectConvConfigArray(T func_f32, T func_f16)
Constructor.
Basic container for the OpenCL indirect convolution configuration functions.
IClIndirectConvKernelConfig(GPUTarget arch)
Constructor.
T get_function(DataType data_type)
Method to return the indirect convolution configuration function based on data type.
static constexpr size_t DT_F32
Alias for F32 index.
GPUTarget
Available GPU Targets.
virtual DirectConvComputeKernelInfo configure(const ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info)=0
This method returns the DirectConvComputeKernelInfo for the given inputs.
Copyright (c) 2017-2024 Arm Limited.
@ F16
16-bit floating-point number
static constexpr size_t DT_F16
Alias for F16 index.
Store the tensor's metadata.
@ F32
32-bit floating-point number
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(IClIndirectConvKernelConfig)
DataType
Available data types.
Compute descriptor used by the direct convolution kernel.