diff options
Diffstat (limited to 'src/backends/reference/workloads/ReverseV2Impl.hpp')
-rw-r--r-- | src/backends/reference/workloads/ReverseV2Impl.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/ReverseV2Impl.hpp b/src/backends/reference/workloads/ReverseV2Impl.hpp index bc1fe1d432..59407d4a4e 100644 --- a/src/backends/reference/workloads/ReverseV2Impl.hpp +++ b/src/backends/reference/workloads/ReverseV2Impl.hpp @@ -13,9 +13,10 @@ namespace armnn { -void ReverseV2(const ReverseV2Descriptor& params, - const TensorInfo& inputInfo, +void ReverseV2(const TensorInfo& inputInfo, + const TensorInfo& axisInfo, Decoder<float>& inputDecoder, + Decoder<int>& axisDecoder, Encoder<float>& outputEncoder); -} // namespace armnn
\ No newline at end of file +} // namespace armnn |