blob: be255c285f43c8d8ebee80176b788a5a8f6bccc7 [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
Sadik Armagan310d8ff2019-07-11 10:53:38 +010015armnn::TensorShape InferPadOutputShape(const armnn::TensorShape& inputShape,
16 const std::vector<std::pair<unsigned int, unsigned int>>& padList);
17
Aron Virginas-Tarf03fcf02019-07-09 17:44:24 +010018armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape);
19
Sadik Armagan5e9521c2019-07-12 13:55:57 +010020armnn::TensorShape InferSubOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape);
21
Aron Virginas-Tarf03fcf02019-07-09 17:44:24 +010022} // namespace armnn_driver
23
24