diff options
Diffstat (limited to 'delegate/opaque/src/LogicalBinary.hpp')
-rw-r--r-- | delegate/opaque/src/LogicalBinary.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/delegate/opaque/src/LogicalBinary.hpp b/delegate/opaque/src/LogicalBinary.hpp index 44a443bb4d..3bac72bf56 100644 --- a/delegate/opaque/src/LogicalBinary.hpp +++ b/delegate/opaque/src/LogicalBinary.hpp @@ -119,7 +119,9 @@ TfLiteStatus VisitLogicalBinaryOperator(DelegateData& delegateData, return isSupported ? kTfLiteOk : kTfLiteError; } - armnn::IConnectableLayer* logicalBinaryLayer = delegateData.m_Network->AddLogicalBinaryLayer(desc); + auto layerName = GetName(desc.m_Operation, nodeIndex); + armnn::IConnectableLayer* logicalBinaryLayer = delegateData.m_Network->AddLogicalBinaryLayer(desc, + layerName.c_str()); logicalBinaryLayer->SetBackendId(setBackend); ARMNN_ASSERT(logicalBinaryLayer != nullptr); @@ -129,7 +131,8 @@ TfLiteStatus VisitLogicalBinaryOperator(DelegateData& delegateData, auto inputsTensorsProcess = ProcessInputs(logicalBinaryLayer, delegateData, tfLiteContext, - tfLiteNode); + tfLiteNode, + nodeIndex); if (inputsTensorsProcess == kTfLiteError) { return inputsTensorsProcess; |