diff options
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
-rw-r--r-- | src/armnnTfParser/TfParser.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index b646437f36..45c039bb15 100644 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -333,6 +333,7 @@ const std::map<std::string, TfParser::OperationParsingFunction> TfParser::ms_Ope { "DepthwiseConv2dNative", &TfParser::ParseDepthwiseConv2D }, { "ExpandDims", &TfParser::ParseExpandDims }, { "FusedBatchNorm", &TfParser::ParseFusedBatchNorm }, + { "Greater", &TfParser::ParseGreater}, { "ConcatV2", &TfParser::ParseConcat }, { "LRN", &TfParser::ParseLrn }, { "MatMul", &TfParser::ParseMatMul }, @@ -1592,6 +1593,18 @@ ParsedTfOperationPtr TfParser::ProcessElementwiseLayer( return std::make_unique<SingleLayerParsedTfOperation>(this, nodeDef, layer); } +ParsedTfOperationPtr TfParser::ParseGreater(const tensorflow::NodeDef& nodeDef, + const tensorflow::GraphDef& graphDef) +{ + std::pair<armnn::IOutputSlot*, armnn::IOutputSlot*> inputLayers = ProcessElementwiseInputSlots(nodeDef, "Greater"); + IOutputSlot* input0Slot = inputLayers.first; + IOutputSlot* input1Slot = inputLayers.second; + + IConnectableLayer* const layer = m_Network->AddGreaterLayer(nodeDef.name().c_str()); + + return ProcessElementwiseLayer(input0Slot, input1Slot, layer, nodeDef); +} + ParsedTfOperationPtr TfParser::ParseEqual(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef) { |