diff options
Diffstat (limited to 'src/armnn/layers/OutputLayer.hpp')
| -rw-r--r-- | src/armnn/layers/OutputLayer.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/layers/OutputLayer.hpp b/src/armnn/layers/OutputLayer.hpp index 6315e25da1..fc6a8aa6b2 100644 --- a/src/armnn/layers/OutputLayer.hpp +++ b/src/armnn/layers/OutputLayer.hpp @@ -26,9 +26,10 @@ public: /// @param [in] IsMemoryManaged Determine whether or not to assign a memory manager during creation virtual void CreateTensorHandles(const TensorHandleFactoryRegistry& registry, const IWorkloadFactory& factory, - const bool IsMemoryManaged = true) override + const bool isMemoryManaged = true, + MemorySource memSource = MemorySource::Undefined) override { - IgnoreUnused(registry, factory, IsMemoryManaged); + IgnoreUnused(registry, factory, isMemoryManaged, memSource); } /// Creates a dynamically-allocated copy of this layer. |