From 0e4e43956687bda724339a5243a82f53b7154a5c Mon Sep 17 00:00:00 2001
From: Finn Williams <Finn.Williams@arm.com>
Date: Wed, 31 Jul 2019 10:56:27 +0100
Subject: IVGCVSW-3583 Correct data layout index for ConvertBatchToSpaceNd

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: Ibd683c147b6f2fb966a8f90105c9213880100cf8
---
 ConversionUtils.hpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index 946bc95d..2b2a8ce4 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -1802,13 +1802,12 @@ bool ConvertSub(const Operation& operation, const Model& model, ConversionData&
 
 template<typename HalPolicy,
          typename HalOperation   = typename HalPolicy::Operation,
+         typename HalOperand     = typename HalPolicy::Operand,
          typename HalModel       = typename HalPolicy::Model>
 bool ConvertBatchToSpaceNd(const HalOperation& operation,
                            const HalModel& model,
                            ConversionData& data)
 {
-    using HalOperand     = typename HalPolicy::Operand;
-    using HalOperandType = typename HalPolicy::OperandType;
 
     LayerInputHandle input = ConvertToLayerInputHandle<HalPolicy>(operation, 0, model, data);
     if (!input.IsValid())
@@ -1861,7 +1860,7 @@ bool ConvertBatchToSpaceNd(const HalOperation& operation,
 
     if (Is12Operand(*output))
     {
-        batchToSpaceNdDesc.m_DataLayout = OptionalDataLayout<HalPolicy>(operation, 3, model, data);
+        batchToSpaceNdDesc.m_DataLayout = OptionalDataLayout<HalPolicy>(operation, 2, model, data);
     }
     // Setting crops to 0,0 0,0 as it is not supported in Android NN API
     batchToSpaceNdDesc.m_Crops = {{0, 0}, {0, 0}};
-- 
cgit v1.2.1