24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_KERNELS_CPUSCALEKERNEL_H
25 #define ACL_SRC_CPU_KERNELS_CPUSCALEKERNEL_H
43 using ScaleKernelPtr = std::add_pointer<void(
const ITensor *,
92 const char *
name()
const override;
104 ScaleKernelPtr _nchw_func{
nullptr};
108 float _sampling_offset{0};
109 bool _align_corners{
false};
111 ScaleKernelPtr _run_method{
nullptr};
117 #endif // ACL_SRC_CPU_KERNELS_CPUSCALEKERNEL_H
SimpleTensor< float > src
decltype(strategy::transforms) typedef type
Class describing the value of a pixel for any image format.
const char * name() const override
Name of the kernel.
DataLayout
[DataLayout enum definition]
std::add_pointer< bool(const ScaleKernelDataTypeISASelectorData &data)>::type ScaleKernelDataTypeISASelectorDataPtr
InterpolationPolicy
Interpolation method.
Interface for CPU tensor.
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
const ScaleKernelDataTypeISASelectorDataPtr is_selected
Arm(R) Neon(TM) kernel to perform scaling on a tensor.
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
static Status validate(const ITensorInfo *src, const ITensorInfo *dx, const ITensorInfo *dy, const ITensorInfo *offsets, ITensorInfo *dst, const ScaleKernelInfo &info)
Static function to check if given info will lead to a valid configuration.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
BorderMode
Methods available to handle borders.
static const std::vector< ScaleKernel > & get_available_kernels()
void configure(const ITensorInfo *src, const ITensorInfo *dx, const ITensorInfo *dy, const ITensorInfo *offsets, ITensorInfo *dst, const ScaleKernelInfo &info)
Initialise the kernel's inputs, output and interpolation policy.
@ 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(CpuScaleKernel)