24 #ifndef ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTSKERNEL_H 25 #define ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTSKERNEL_H const Window & window() const
The maximum window the kernel can be executed on.
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) ...
const DataLayout data_layout
Store the tensor's metadata.
Common interface for all the OpenCL kernels.
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...
Copyright (c) 2017-2021 Arm Limited.
~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
[DataLayout enum definition]
Describe a multidimensional execution window.