40 constexpr
int block_size = 3;
45 for(uint32_t i = 0; i < num_elements; ++i)
BorderMode
Methods available to handle borders.
T tensor_elem_at(const SimpleTensor< T > &src, Coordinates coord, BorderMode border_mode, T constant_border_value)
Container for 2D border size.
DataType data_type() const override
Data type of the tensor.
TensorShape shape() const override
Shape of the tensor.
const ValidRegion valid_region
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
T x() const
Alias to access the size of the first dimension.
bool is_in_valid_region(const ValidRegion &valid_region, Coordinates coord)
Check if a coordinate is within a valid region.
Coordinates index2coord(const TensorShape &shape, int index)
Convert a linear index into n-dimensional coordinates.
Simple tensor object that stores elements in a consecutive chunk of memory.
int num_channels() const override
Number of channels of the tensor.
SimpleTensor< T > non_maxima_suppression(const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value)
Borders are left undefined.
int num_elements() const override
Number of elements of the tensor.
T y() const
Alias to access the size of the second dimension.
Container for valid region of a window.
ValidRegion shape_to_valid_region(const TensorShape &a_shape, bool border_undefined=false, BorderSize border_size=BorderSize(0))
Create a valid region based on tensor shape, border mode and border size.