aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/SplitterLayer.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2021-05-07 17:52:36 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2021-05-10 10:22:38 +0100
commit0c76b23726a86a46da44c3f52348db53f73ae242 (patch)
tree0e32680ed15ed5157c78d5deeabda2c0ceeeb4a3 /src/armnn/layers/SplitterLayer.hpp
parentf370e312bd26c29c454ea82bb08ddb614f010484 (diff)
downloadarmnn-0c76b23726a86a46da44c3f52348db53f73ae242.tar.gz
IVGCVSW-5818 Enable import on GPU
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: I4e4eb107aa2bfa09625840d738001f33152e6792
Diffstat (limited to 'src/armnn/layers/SplitterLayer.hpp')
-rw-r--r--src/armnn/layers/SplitterLayer.hpp3
1 files changed, 1 insertions, 2 deletions
diff --git a/src/armnn/layers/SplitterLayer.hpp b/src/armnn/layers/SplitterLayer.hpp
index 075b136da9..f90696b1ad 100644
--- a/src/armnn/layers/SplitterLayer.hpp
+++ b/src/armnn/layers/SplitterLayer.hpp
@@ -26,8 +26,7 @@ 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,
- MemorySource memSource = MemorySource::Undefined) override;
+ const bool IsMemoryManaged = true) override;
/// Creates a dynamically-allocated copy of this layer.
/// @param [in] graph The graph into which this layer is being cloned.