24.02.1
|
Go to the documentation of this file.
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;
SimpleTensor< float > src
int coord2index(const TensorShape &shape, const Coordinates &coord)
Linearise the given coordinate.
T z() const
Alias to access the size of the third dimension.
size_t total_size() const
Collapses all dimensions to a single linear total size.
T x() const
Alias to access the size of the first dimension.
Simple tensor object that stores elements in a consecutive chunk of memory.
const unsigned int num_groups
Copyright (c) 2017-2024 Arm Limited.
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)