aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSaoirse Stewart <saoirse.stewart@arm.com>2019-02-28 11:32:41 +0000
committerSadik Armagan <sadik.armagan@arm.com>2019-02-28 12:19:24 +0000
commit120196d394eb5af4b28f78396d1a225ca47cf274 (patch)
tree24cfc616ce0a945e8b3838b6ccf4f5946f90a9d8
parent91c0eff2ff4ff384e013cb69cac1e07e28b9e2b1 (diff)
downloadarmnn-branches/armnn_19_02.tar.gz
IVGCVSW-2598 Update attribute to use num_splitv19.02branches/armnn_19_02
Change-Id: I5de0e0a2de9241498213d274a6bf2a62ba7da7cc Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
-rwxr-xr-xsrc/armnnTfParser/TfParser.cpp4
-rw-r--r--src/armnnTfParser/test/Split.cpp2
2 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp
index 1e304cbfd7..8e57e56917 100755
--- a/src/armnnTfParser/TfParser.cpp
+++ b/src/armnnTfParser/TfParser.cpp
@@ -2637,8 +2637,8 @@ ParsedTfOperationPtr TfParser::ParseSplit(const tensorflow::NodeDef& nodeDef,
% 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();
diff --git a/src/armnnTfParser/test/Split.cpp b/src/armnnTfParser/test/Split.cpp
index 87cd6544c9..bf42bf7c5d 100644
--- a/src/armnnTfParser/test/Split.cpp
+++ b/src/armnnTfParser/test/Split.cpp
@@ -106,7 +106,7 @@ struct SplitFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::IT
}
m_Prototext += R"(
attr {
- key: "num_or_size_splits"
+ key: "num_split"
value {
i: 2
}