44 const size_t src_width = src.
shape().x();
45 const size_t src_height = src.
shape().y();
51 #pragma omp parallel for collapse(3) 55 for(
size_t x = 0; x < src_width; ++x)
57 for(
size_t y = 0; y < src_height; ++y)
59 const int dst_idx = y + x * src_height +
b * src_height * src_width;
68 #pragma omp parallel for collapse(4) 74 for(
size_t x = 0; x < src_width; ++x)
76 for(
size_t y = 0; y < src_height; ++y)
78 const int dst_idx = y + x * src_height + g * src_height * src_width +
b * src_height * src_width *
num_groups;
DataType data_type() const override
Data type of the tensor.
TensorShape shape() const override
Shape of the tensor.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
int coord2index(const TensorShape &shape, const Coordinates &coord)
Linearise the given coordinate.
T x() const
Alias to access the size of the first dimension.
T z() const
Alias to access the size of the third dimension.
const unsigned int num_groups
size_t total_size() const
Collapses all dimensions to a single linear total size.
Simple tensor object that stores elements in a consecutive chunk of memory.
T y() const
Alias to access the size of the second dimension.
SimpleTensor< T > col2im(const SimpleTensor< T > &src, const TensorShape &dst_shape, unsigned int num_groups)