From fc884b4141a28fbd3c62f665341ec88158fcd332 Mon Sep 17 00:00:00 2001
From: Finn Williams <Finn.Williams@Arm.com>
Date: Thu, 11 Jun 2020 17:35:44 +0100
Subject: Add support for Rank and scalar tensors

!armnn:3330

Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: Icc429d9fabb570193d12bffef0e00dda7b51032f
---
 ConversionUtils.hpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'ConversionUtils.hpp')

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,
-- 
cgit v1.2.1