diff options
author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-28 11:32:41 +0000 |
---|---|---|
committer | Saoirse Stewart Arm <saoirse.stewart@arm.com> | 2019-02-28 11:37:17 +0000 |
commit | 315258e6a2692eef2727326508577e5c73194e85 (patch) | |
tree | dba19e88a9a137b00f3b3ff400103894460b3305 /src/armnnTfParser/TfParser.cpp | |
parent | 6b4ed983aec66a979fbf206291d387bad0eeec9d (diff) | |
download | armnn-315258e6a2692eef2727326508577e5c73194e85.tar.gz |
IVGCVSW-2598 Update attribute to use num_split
Change-Id: I5de0e0a2de9241498213d274a6bf2a62ba7da7cc
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 4 |
1 files changed, 2 insertions, 2 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(); |