diff options
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
| -rw-r--r-- | src/armnnTfParser/TfParser.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 9a7d7827c3..2b809419de 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -187,6 +187,12 @@ private: std::pair<armnn::IOutputSlot*, armnn::IOutputSlot*> ProcessElementwiseInputSlots( const tensorflow::NodeDef& nodeDef, const std::string& layerName); + ParsedTfOperationPtr ProcessComparisonLayer( + armnn::IOutputSlot* input0Slot, + armnn::IOutputSlot* input1Slot, + armnn::IConnectableLayer* const layer, + const tensorflow::NodeDef& nodeDef); + ParsedTfOperationPtr ProcessElementwiseLayer( armnn::IOutputSlot* input0Slot, armnn::IOutputSlot* input1Slot, |