2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
8 #include <armnn/TypesUtils.hpp>
12 armnn::TensorShape GetTensorShape(unsigned int numberOfBatches,
13 unsigned int numberOfChannels,
16 const armnn::DataLayout dataLayout);
18 armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches,
19 unsigned int numberOfChannels,
22 const armnn::DataLayout dataLayout,
23 const armnn::DataType dataType);
25 } // namespace armnnUtils