24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_KERNELS_CPUIM2COLKERNEL_H
25 #define ACL_SRC_CPU_KERNELS_CPUIM2COLKERNEL_H
89 unsigned int input_pad_right = 0);
103 unsigned int input_pad_right = 0);
107 const char *
name()
const override;
115 size_t get_mws(
const CPUInfo &platform,
size_t thread_count)
const override;
122 using Im2ColFunctionPtr = void (*)(
const ITensor *
src,
127 std::pair<unsigned int, unsigned int> convolved_dims,
128 const Size2D &kernel_dims,
130 uint32_t input_pad_right,
133 Im2ColFunctionPtr _func{
nullptr};
134 std::pair<unsigned int, unsigned int> _convolved_dims{};
136 unsigned int _kernel_width{0};
137 unsigned int _kernel_height{0};
138 unsigned int _input_pad_right{0};
139 bool _has_bias{
false};
146 #endif // ACL_SRC_CPU_KERNELS_CPUIM2COLKERNEL_H
CpuIm2ColKernel()=default
Default constructor.
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
SimpleTensor< float > src
DataLayout
[DataLayout enum definition]
Class for specifying the size of an image or rectangle.
Interface for CPU tensor.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation=Size2D(1U, 1U), unsigned int num_groups=1, unsigned int input_pad_right=0)
Static function to check if given info will lead to a valid configuration.
Interface for the im2col reshape kernel.
constexpr auto data_layout
const char * name() const override
Name of the kernel.
size_t get_mws(const CPUInfo &platform, size_t thread_count) const override
Return minimum workload size of the relevant kernel.
void configure(const ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation=Size2D(1U, 1U), unsigned int num_groups=1, unsigned int input_pad_right=0)
Set the input and output of the kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
const unsigned int num_groups
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
@ UNKNOWN
Unknown data layout.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuIm2ColKernel)