diff options
Diffstat (limited to 'src/backends/reference/workloads/PreluImpl.cpp')
-rw-r--r-- | src/backends/reference/workloads/PreluImpl.cpp | 8 |
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(); |