diff options
Diffstat (limited to 'src/backends/reference/workloads/PreluImpl.hpp')
-rw-r--r-- | src/backends/reference/workloads/PreluImpl.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/PreluImpl.hpp b/src/backends/reference/workloads/PreluImpl.hpp index 9299b1c7f7..0b3d3b08e5 100644 --- a/src/backends/reference/workloads/PreluImpl.hpp +++ b/src/backends/reference/workloads/PreluImpl.hpp @@ -13,7 +13,9 @@ 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); |