24 #ifndef __ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTSKERNEL_H__ 25 #define __ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTSKERNEL_H__
CLConvertFullyConnectedWeightsKernel()
Default constructor.
Interface to convert the 2D Fully Connected weights from NCHW to NHWC or vice versa.
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue...
CLConvertFullyConnectedWeightsKernel & operator=(const CLConvertFullyConnectedWeightsKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Store the tensor's metadata.
Common interface for all the OpenCL kernels.
This file contains all available output stages for GEMMLowp on OpenCL.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, DataLayout data_layout)
Static function to check if given info will lead to a valid configuration of CLConvertFullyConnectedW...
~CLConvertFullyConnectedWeightsKernel()=default
Default destructor.
void configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, DataLayout data_layout)
Set the input and output tensor.
Interface for OpenCL tensor.
DataLayout
Supported tensor data layouts.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.