diff options
-rw-r--r-- | src/armnn/Graph.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index fe6a7c8afa..d44e43c2cb 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -199,6 +199,11 @@ Status Graph::AllocateDynamicBuffers() { handleReferenceCounts[tensorHandle] = numConnections; tensorHandle->Manage(); + if (handleReferenceCounts[tensorHandle] == 0u) + { + // if nobody consumes this tensor we call Allocate() + tensorHandle->Allocate(); + } } else { |