24.02.1
|
Valhall based OpenCL direct convolution configuration. More...
#include <ClDirectConvDefaultConfigValhall.h>
Public Member Functions | |
ClDirectConvDefaultConfigValhall (GPUTarget gpu) | |
Constructor. More... | |
DirectConvComputeKernelInfo | configure (const ITensorInfo *src, const ITensorInfo *wei, const PadStrideInfo &conv_info) override |
This method returns the DirectConvComputeKernelInfo for the given inputs. More... | |
Public Member Functions inherited from IClDirectConvKernelConfig | |
IClDirectConvKernelConfig (GPUTarget arch) | |
Constructor. More... | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (IClDirectConvKernelConfig) | |
virtual | ~IClDirectConvKernelConfig ()=default |
Virtual destructor. More... | |
Valhall based OpenCL direct convolution configuration.
Definition at line 34 of file ClDirectConvDefaultConfigValhall.h.
Constructor.
[in] | gpu | GPU target |
Definition at line 41 of file ClDirectConvDefaultConfigValhall.cpp.
|
overridevirtual |
This method returns the DirectConvComputeKernelInfo for the given inputs.
[in] | src | Source tensor (activation tensor) |
[in] | wei | Weights tensor |
[in] | conv_info | Convolution info |
Implements IClDirectConvKernelConfig.
Definition at line 45 of file ClDirectConvDefaultConfigValhall.cpp.
References ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::conv_info, arm_compute::G57, arm_compute::G78, ClDirectConvConfigArray< T >::get_function(), and arm_compute::test::validation::src.