diff options
Diffstat (limited to 'delegate/classic/src/ReverseV2.hpp')
| -rw-r--r-- | delegate/classic/src/ReverseV2.hpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/delegate/classic/src/ReverseV2.hpp b/delegate/classic/src/ReverseV2.hpp index d49d20b5c1..64941f7c5c 100644 --- a/delegate/classic/src/ReverseV2.hpp +++ b/delegate/classic/src/ReverseV2.hpp @@ -109,8 +109,6 @@ TfLiteStatus VisitReverseV2Operator(DelegateData& delegateData, } } - std::string layerName("ReverseV2"); - const auto maxDimension = 4; const auto axisTensorNumValues = static_cast<unsigned int>(tfLiteAxisTensor.dims->size); @@ -135,13 +133,14 @@ TfLiteStatus VisitReverseV2Operator(DelegateData& delegateData, outputTensorInfo); } + auto layerName = GetLayerName(armnn::LayerType::ReverseV2, nodeIndex); armnn::IConnectableLayer* reverseV2Layer = delegateData.m_Network->AddReverseV2Layer(layerName.c_str()); armnn::IOutputSlot& outputSlot = reverseV2Layer->GetOutputSlot(0); outputSlot.SetTensorInfo(outputTensorInfo); // Try to connect the Constant Inputs if there are any - if(ProcessInputs(reverseV2Layer, delegateData, tfLiteContext, tfLiteNode) != kTfLiteOk ) + if (ProcessInputs(reverseV2Layer, delegateData, tfLiteContext, tfLiteNode, nodeIndex) != kTfLiteOk) { return kTfLiteError; } |