diff options
author | Finn Williams <Finn.Williams@Arm.com> | 2020-06-11 17:35:44 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@Arm.com> | 2020-07-06 19:07:52 +0100 |
commit | fc884b4141a28fbd3c62f665341ec88158fcd332 (patch) | |
tree | df18555baced3f4d917a78d24d87c5e2d1778f83 /ConversionUtils.hpp | |
parent | 00b586b1f3106a4e6970ca7feacb1cc1892d107e (diff) | |
download | android-nn-driver-fc884b4141a28fbd3c62f665341ec88158fcd332.tar.gz |
Add support for Rank and scalar tensors
!armnn:3330
Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: Icc429d9fabb570193d12bffef0e00dda7b51032f
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 0fbd4e4d..5dc9993d 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -1451,14 +1451,16 @@ bool SetupAndTrackLayerOutputSlot(const HalOperation& operation, uint32_t outputIndex, armnn::IConnectableLayer& layer, const HalModel& model, - ConversionData& data) + ConversionData& data, + const armnn::TensorInfo* overrideOutputInfo = nullptr) { return SetupAndTrackLayerOutputSlot<HalPolicy>(operation, outputIndex, layer, outputIndex, model, - data); + data, + overrideOutputInfo); } template<typename HalPolicy, |