aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/PreluImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/PreluImpl.cpp')
-rw-r--r--src/backends/reference/workloads/PreluImpl.cpp8
1 files changed, 3 insertions, 5 deletions
diff --git a/src/backends/reference/workloads/PreluImpl.cpp b/src/backends/reference/workloads/PreluImpl.cpp
index 458025bb0a..6df259fa4d 100644
--- a/src/backends/reference/workloads/PreluImpl.cpp
+++ b/src/backends/reference/workloads/PreluImpl.cpp
@@ -10,15 +10,13 @@
namespace armnn
{
-void PreluImpl(const PreluQueueDescriptor& data,
+void PreluImpl(const TensorInfo& inputInfo,
+ const TensorInfo& alphaInfo,
+ const TensorInfo& outputInfo,
Decoder<float>& inputData,
Decoder<float>& alphaData,
Encoder<float>& outputData)
{
- const TensorInfo& inputInfo = GetTensorInfo(data.m_Inputs[0]);
- const TensorInfo& alphaInfo = GetTensorInfo(data.m_Inputs[1]);
- const TensorInfo& outputInfo = GetTensorInfo(data.m_Outputs[0]);
-
const TensorShape& inputShape = inputInfo.GetShape();
const TensorShape& alphaShape = alphaInfo.GetShape();
const TensorShape& outputShape = outputInfo.GetShape();