diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-05 13:33:33 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-05 13:33:33 +0100 |
commit | 725728e7d46c1e672bbdc72cf86e22db6fb210ee (patch) | |
tree | b93901e1308f6b01bb41577237f05864518058a5 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | bd22c7d8d71bb9d6fdebcd07a472d66c7616abad (diff) | |
download | armnn-725728e7d46c1e672bbdc72cf86e22db6fb210ee.tar.gz |
IVGCVSW-6938 Do not add Floor when FloorDiv is int32 in Tfliteparser
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7ce633a66e2ecb72a9cdd1bff690c4195a9a449f
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 7fe954d901..5f71ebcff6 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -4341,12 +4341,21 @@ armnn::IConnectableLayer* TfLiteParserImpl::AddFusedActivationLayer(armnn::IConn armnn::IConnectableLayer* TfLiteParserImpl::AddFusedFloorLayer(armnn::IConnectableLayer* prevLayer, unsigned int outputSlot) { + + auto& prevOutputSlot = prevLayer->GetOutputSlot(outputSlot); + DataType dataType = prevOutputSlot.GetTensorInfo().GetDataType(); + + if (dataType == DataType::Signed32) + { + return prevLayer; + } + std::string layerName = prevLayer->GetName(); IConnectableLayer* floorLayer = m_Network->AddFloorLayer(layerName.c_str()); - auto & prevOutputSlot = prevLayer->GetOutputSlot(outputSlot); prevOutputSlot.Connect(floorLayer->GetInputSlot(0)); floorLayer->GetOutputSlot(0).SetTensorInfo(prevOutputSlot.GetTensorInfo()); + return floorLayer; } |