aboutsummaryrefslogtreecommitdiff
path: root/delegate/classic/src/Resize.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/classic/src/Resize.hpp')
-rw-r--r--delegate/classic/src/Resize.hpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/delegate/classic/src/Resize.hpp b/delegate/classic/src/Resize.hpp
index 32c7f46b9f..cede32b6bf 100644
--- a/delegate/classic/src/Resize.hpp
+++ b/delegate/classic/src/Resize.hpp
@@ -185,14 +185,14 @@ TfLiteStatus VisitResizeOperator(DelegateData& delegateData,
}
- armnn::IConnectableLayer* resizeLayer = nullptr;
- resizeLayer = delegateData.m_Network->AddResizeLayer(desc, layerName.c_str());
+ auto resizeName = GetLayerName(armnn::LayerType::Resize, nodeIndex);
+ armnn::IConnectableLayer* resizeLayer = delegateData.m_Network->AddResizeLayer(desc, resizeName.c_str());
armnn::IOutputSlot& outputSlot = resizeLayer->GetOutputSlot(0);
outputSlot.SetTensorInfo(outputTensorInfo);
// try to connect the Constant Inputs if there are any
- if(ProcessInputs(resizeLayer,delegateData, tfLiteContext, tfLiteNode) != kTfLiteOk )
+ if (ProcessInputs(resizeLayer, delegateData, tfLiteContext, tfLiteNode, nodeIndex) != kTfLiteOk)
{
return kTfLiteError;
}