24 #ifndef ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTS_H 25 #define ARM_COMPUTE_CLCONVERTFULLYCONNECTEDWEIGHTS_H 34 class CLCompileContext;
35 class CLConvertFullyConnectedWeightsKernel;
74 namespace weights_transformations
83 _output.allocator()->allocate();
91 _output.allocator()->free();
124 _func.configure(compile_context, input, &_output, original_input_shape, data_layout);
128 static constexpr uint32_t _uid = 0x5;
const DataLayout data_layout
static CLKernelLibrary & get()
Access the KernelLibrary singleton.
Store the tensor's metadata.
Copyright (c) 2017-2021 Arm Limited.
Basic function to run CLConvertFullyConnectedWeightsKernel.
Basic interface for functions which have a single OpenCL kernel.
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...
Interface for OpenCL tensor.
Manages all the OpenCL kernels compilation and caching, provides accessors for the OpenCL Context...
void configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, DataLayout data_layout)
Initialize the function.
DataLayout
[DataLayout enum definition]
Basic implementation of the OpenCL tensor interface.