24 #ifndef ARM_COMPUTE_CLDEPTHWISECONVOLUTIONNHWCKERNEL3x3_H 25 #define ARM_COMPUTE_CLDEPTHWISECONVOLUTIONNHWCKERNEL3x3_H 60 const ICLTensor *output_multipliers =
nullptr,
const ICLTensor *output_shifts =
nullptr)
override;
81 const ICLTensor *output_multipliers =
nullptr,
const ICLTensor *output_shifts =
nullptr)
override;
110 unsigned int _num_planes_processed_per_iteration;
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue...
const Window & window() const
The maximum window the kernel can be executed on.
Interface for the kernel to run a 3x3 depthwise convolution on a tensor.
Container for 2D border size.
Store the tensor's metadata.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, ActivationLayerInfo act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1U, 1U), const ITensorInfo *output_multipliers=nullptr, const ITensorInfo *output_shifts=nullptr)
Static function to check if given info will lead to a valid configuration of CLDepthwiseConvolutionLa...
BorderSize border_size() const override
The size of the border for that kernel.
void configure(const ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, ActivationLayerInfo act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1U, 1U), const ICLTensor *output_multipliers=nullptr, const ICLTensor *output_shifts=nullptr) override
Default move assignment operator.
Padding and stride information class.
Interface for the kernel to run a 3x3 depthwise convolution on a tensor when the data layout is NHWC...
CLDepthwiseConvolutionLayer3x3NHWCKernel()
Default constructor.
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
Describe a multidimensional execution window.