blob: dac4a193c0ba02614db3c52c7e766839a90d4bd7 [file] [log] [blame]
Aron Virginas-Tarf03fcf02019-07-09 17:44:24 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include <armnn/ArmNN.hpp>
9
10namespace armnn_driver
11{
12
Aron Virginas-Tar366e0a62019-07-10 13:01:41 +010013bool IsDynamicOutput(const armnn::TensorInfo& outputInfo);
14
Narumol Prangnawarat95b1ef62019-07-15 12:02:20 +010015armnn::TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape,
16 const armnn::TensorShape& input1Shape);
17
Sadik Armagan310d8ff2019-07-11 10:53:38 +010018armnn::TensorShape InferPadOutputShape(const armnn::TensorShape& inputShape,
19 const std::vector<std::pair<unsigned int, unsigned int>>& padList);
20
Aron Virginas-Tarf03fcf02019-07-09 17:44:24 +010021armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape);
22
Sadik Armagan5e9521c2019-07-12 13:55:57 +010023armnn::TensorShape InferSubOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape);
24
Aron Virginas-Tarf03fcf02019-07-09 17:44:24 +010025} // namespace armnn_driver
26
27