diff options
Diffstat (limited to 'src/backends/reference/workloads/Stack.hpp')
-rw-r--r-- | src/backends/reference/workloads/Stack.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/Stack.hpp b/src/backends/reference/workloads/Stack.hpp index cd86d41552..2f02a4c644 100644 --- a/src/backends/reference/workloads/Stack.hpp +++ b/src/backends/reference/workloads/Stack.hpp @@ -15,6 +15,8 @@ namespace armnn void Stack (const StackQueueDescriptor& data, std::vector<std::unique_ptr<Decoder<float>>>& inputs, - Encoder<float>& output); + Encoder<float>& output, + const TensorInfo& inputInfo, + const TensorInfo& outputInfo); } // namespace armnn |