24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NEON_CROP_RESIZE_H
25 #define ARM_COMPUTE_NEON_CROP_RESIZE_H
84 float extrapolation_value = 0);
110 float extrapolation_value);
119 std::vector<std::unique_ptr<NECropKernel>>
_crop;
120 std::vector<std::unique_ptr<NEScale>>
_scale;
Base class for all functions.
std::vector< std::unique_ptr< Tensor > > _scaled_results
InterpolationPolicy
Interpolation method.
Interface for CPU tensor.
static Status validate(const ITensorInfo *input, const ITensorInfo *boxes, const ITensorInfo *box_ind, const ITensorInfo *output, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value)
Static function to check if given info will lead to a valid configuration of NESlice.
float _extrapolation_value
~NECropResize()
Default destructor.
void configure(const ITensor *input, const ITensor *boxes, const ITensor *box_ind, ITensor *output, Coordinates2D crop_size, InterpolationPolicy method=InterpolationPolicy::BILINEAR, float extrapolation_value=0)
Configure kernel.
@ BILINEAR
Output values are defined by bilinear interpolation between the pixels.
NECropResize()
Default constructor.
InterpolationPolicy _method
std::vector< std::unique_ptr< NEScale > > _scale
Function to perform cropping and resizing.
std::vector< std::unique_ptr< NECropKernel > > _crop
std::vector< std::unique_ptr< Tensor > > _crop_results
Copyright (c) 2017-2024 Arm Limited.
NECropResize & operator=(const NECropResize &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
Store the tensor's metadata.