diff options
Diffstat (limited to 'src')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 4 | ||||
-rw-r--r-- | src/armnnTfParser/test/Split.cpp | 2 |
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 } |