diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-17 17:02:12 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-17 18:46:28 +0000 |
| commit | ef6f300928e54cbe6bbfb9dfeb26c2db56ee5dc9 (patch) | |
| tree | 7f9be0980ff5cede5ff8b25fced44f3c5cc0bd70 /src/armnn/layers/ConcatLayer.hpp | |
| parent | 35c31c04a9805082be32474211dcd62b7629ed2b (diff) | |
| download | armnn-ef6f300928e54cbe6bbfb9dfeb26c2db56ee5dc9.tar.gz | |
IVGCVSW-5114 Enable memory import in TfLiteYoloV3Big App
* Enable memory import in TfLiteYoloV3Big App
* Add isMemoryManaged flag to Concat and Splitter layers
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I7e00f5da2a016c09d480b744fb17ea5611af8365
Diffstat (limited to 'src/armnn/layers/ConcatLayer.hpp')
| -rw-r--r-- | src/armnn/layers/ConcatLayer.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/ConcatLayer.hpp b/src/armnn/layers/ConcatLayer.hpp index eaa5c15a9c..3d9ba1815e 100644 --- a/src/armnn/layers/ConcatLayer.hpp +++ b/src/armnn/layers/ConcatLayer.hpp @@ -56,7 +56,7 @@ protected: private: template <typename FactoryType> - void CreateTensors(const TensorHandleFactoryRegistry& registry, const FactoryType& factory); + void CreateTensors(const TensorHandleFactoryRegistry& registry, const FactoryType& factory, bool isMemoryManaged); }; |