24.02.1
|
Basic interface for the Direct convolution kernel configuration. More...
#include <IClDirectConvKernelConfig.h>
Public Member Functions | |
IClDirectConvKernelConfig (GPUTarget arch) | |
Constructor. More... | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (IClDirectConvKernelConfig) | |
virtual | ~IClDirectConvKernelConfig ()=default |
Virtual destructor. More... | |
virtual DirectConvComputeKernelInfo | configure (const ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info)=0 |
This method returns the DirectConvComputeKernelInfo for the given inputs. More... | |
Basic interface for the Direct convolution kernel configuration.
Definition at line 88 of file IClDirectConvKernelConfig.h.
|
inline |
Constructor.
[in] | arch | GPU target |
Definition at line 95 of file IClDirectConvKernelConfig.h.
|
virtualdefault |
Virtual destructor.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | IClDirectConvKernelConfig | ) |
|
pure virtual |
This method returns the DirectConvComputeKernelInfo for the given inputs.
[in] | src | Source tensor (activation tensor) |
[in] | wei | Weights tensor |
[in] | conv_info | Convolution info |
Implemented in ClDirectConvDefaultConfigBifrost, and ClDirectConvDefaultConfigValhall.