21.02
|
#include "FlattenLayer.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2021 Arm Limited. | |
arm_compute::test | |
arm_compute::test::validation | |
arm_compute::test::validation::reference | |
Functions | |
template<typename T > | |
SimpleTensor< T > | flatten_layer (const SimpleTensor< T > &src, const TensorShape &shape_flatten) |
template SimpleTensor< float > | flatten_layer (const SimpleTensor< float > &src, const TensorShape &shape_flatten) |
template SimpleTensor< half > | flatten_layer (const SimpleTensor< half > &src, const TensorShape &shape_flatten) |