commit 120196d394eb5af4b28f78396d1a225ca47cf274
author Saoirse Stewart <saoirse.stewart@arm.com> Thu Feb 28 11:32:41 2019 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Thu Feb 28 12:19:24 2019 +0000
tree 24cfc616ce0a945e8b3838b6ccf4f5946f90a9d8
parent 91c0eff2ff4ff384e013cb69cac1e07e28b9e2b1

IVGCVSW-2598 Update attribute to use num_split

Change-Id: I5de0e0a2de9241498213d274a6bf2a62ba7da7cc
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>

src/armnnTfParser/TfParser.cpp

diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp
index 1e304cb..8e57e56 100755
--- a/src/armnnTfParser/TfParser.cpp
+++ b/src/armnnTfParser/TfParser.cpp
@@ -2637,8 +2637,8 @@
                     % CHECK_LOCATION().AsString()));
     }
 
-    // As Armnn only supports splitter outputs of the same shape, therefore num_splits will be limited to an integer.
-    uint32_t num_split = ReadMandatoryNodeUint32Attribute(nodeDef, "num_or_size_splits");
+    // As Armnn only supports splitter outputs of the same shape, therefore num_split will be limited to an integer.
+    uint32_t num_split = ReadMandatoryNodeUint32Attribute(nodeDef, "num_split");
 
     IOutputSlot& inputSlot = inputs[1 - index].m_IndexedValue->ResolveArmnnOutputSlot(inputs[1 - index].m_Index);
     TensorInfo inputTensorInfo = inputSlot.GetTensorInfo();

src/armnnTfParser/test/Split.cpp

diff --git a/src/armnnTfParser/test/Split.cpp b/src/armnnTfParser/test/Split.cpp
index 87cd654..bf42bf7 100644
--- a/src/armnnTfParser/test/Split.cpp
+++ b/src/armnnTfParser/test/Split.cpp
@@ -106,7 +106,7 @@
         }
         m_Prototext += R"(
           attr {
-            key: "num_or_size_splits"
+            key: "num_split"
             value {
               i: 2
             }