diff options
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index 76d25d1d05..d085ed84e3 100755 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -1862,7 +1862,8 @@ ParsedTfOperationPtr TfParser::ParseGreater(const tensorflow::NodeDef& nodeDef, IOutputSlot* input0Slot = inputLayers.first; IOutputSlot* input1Slot = inputLayers.second; - IConnectableLayer* const layer = m_Network->AddGreaterLayer(nodeDef.name().c_str()); + ComparisonDescriptor descriptor(ComparisonOperation::Greater); + IConnectableLayer* const layer = m_Network->AddComparisonLayer(descriptor, nodeDef.name().c_str()); return ProcessComparisonLayer(input0Slot, input1Slot, layer, nodeDef); } @@ -1874,7 +1875,8 @@ ParsedTfOperationPtr TfParser::ParseEqual(const tensorflow::NodeDef& nodeDef, IOutputSlot* input0Slot = inputLayers.first; IOutputSlot* input1Slot = inputLayers.second; - IConnectableLayer* const layer = m_Network->AddEqualLayer(nodeDef.name().c_str()); + ComparisonDescriptor descriptor(ComparisonOperation::Equal); + IConnectableLayer* const layer = m_Network->AddComparisonLayer(descriptor, nodeDef.name().c_str()); return ProcessComparisonLayer(input0Slot, input1Slot, layer, nodeDef); } |