diff options
Diffstat (limited to 'src/backends/backendsCommon/OutputHandler.hpp')
-rw-r--r-- | src/backends/backendsCommon/OutputHandler.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/OutputHandler.hpp b/src/backends/backendsCommon/OutputHandler.hpp index 87ced20706..81768c6f5e 100644 --- a/src/backends/backendsCommon/OutputHandler.hpp +++ b/src/backends/backendsCommon/OutputHandler.hpp @@ -36,8 +36,8 @@ public: /// @brief - Creates tensor handles used by the intermediate tensors. Does not allocate memory. /// @param factory - Factory to be used for handler creation. - void CreateTensorHandles(const IWorkloadFactory& factory); - void CreateTensorHandles(const ITensorHandleFactory& factory); + void CreateTensorHandles(const IWorkloadFactory& factory, const bool IsMemoryManaged = true); + void CreateTensorHandles(const ITensorHandleFactory& factory, const bool IsMemoryManaged = true); /// @brief - Gets the matching TensorInfo for the output. /// @return - References to the output TensorInfo. |