44 const size_t linear_sz = src.
shape().total_size_lower(3);
49 for(
size_t w = 0;
w < group_sz; ++
w)
51 const size_t curr_weight = g * group_sz +
w;
54 for(; i < linear_sz; ++i)
size_t size() const override
Total size of the tensor in bytes.
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.
SimpleTensor< T > weights_reshape(const SimpleTensor< T > &src, const SimpleTensor< T > &biases, const TensorShape &dst_shape, const unsigned int num_groups)
const unsigned int num_groups
Simple tensor object that stores elements in a consecutive chunk of memory.