SimpleTensor< T > flatten_layer(const SimpleTensor< T > &src, const TensorShape &shape_flatten)
DataType data_type() const override
Data type of the tensor.
SimpleTensor< T > copy(const SimpleTensor< T > &src, const TensorShape &output_shape)
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
Simple tensor object that stores elements in a consecutive chunk of memory.
int num_elements() const override
Number of elements of the tensor.
const T * data() const
Constant pointer to the underlying buffer.