diff options
Diffstat (limited to 'src/armnn/Layer.hpp')
| -rw-r--r-- | src/armnn/Layer.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp index 2f5cacc3ce..5ab6b3152f 100644 --- a/src/armnn/Layer.hpp +++ b/src/armnn/Layer.hpp @@ -390,7 +390,7 @@ protected: LayerType* CloneBase(Graph& graph, Params&& ... params) const; // Retrieve the Handles to the constants - using ConstantTensors = std::vector<std::reference_wrapper<std::unique_ptr<ScopedCpuTensorHandle>>>; + using ConstantTensors = std::vector<std::reference_wrapper<std::shared_ptr<ConstCpuTensorHandle>>>; virtual ConstantTensors GetConstantTensorsByRef() {return ConstantTensors(); }; // "Blob" |