24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_WEIGHTSRESHAPE_KERNEL_H
25 #define ARM_COMPUTE_CL_WEIGHTSRESHAPE_KERNEL_H
SimpleTensor< float > src
void run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.
OpenCL kernel to perform reshaping on the weights used by convolution and locally connected layer.
Common interface for all the OpenCL kernels.
const Window & window() const
The maximum window the kernel can be executed on.
const unsigned int num_groups
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
static Status validate(const ITensorInfo *src, const ITensorInfo *biases, const ITensorInfo *dst, unsigned int num_groups=1)
Static function to check if given info will lead to a valid configuration.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(ClWeightsReshapeKernel)
Store the tensor's metadata.
void configure(const ClCompileContext &compile_context, const ITensorInfo *src, const ITensorInfo *biases, ITensorInfo *dst, unsigned int num_groups=1)
Set the input and output of the kernel.