diff options
Diffstat (limited to 'src/armnn/ISubGraphConverter.hpp')
-rw-r--r-- | src/armnn/ISubGraphConverter.hpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/armnn/ISubGraphConverter.hpp b/src/armnn/ISubGraphConverter.hpp index 1d40c6737a..3a6866e102 100644 --- a/src/armnn/ISubGraphConverter.hpp +++ b/src/armnn/ISubGraphConverter.hpp @@ -6,17 +6,21 @@ #pragma once #include <memory> +#include <vector> +#include <functional> namespace armnn { +using CompiledBlobDeleter = std::function<void(const void*)>; +using CompiledBlobPtr = std::unique_ptr<void, CompiledBlobDeleter>; + class ISubGraphConverter { public: - virtual ~ISubGraphConverter() {}; + virtual ~ISubGraphConverter() {} - virtual std::shared_ptr<void> GetOutput() = 0; + virtual std::vector<CompiledBlobPtr> GetOutput() = 0; }; -} - +} // namespace armnn |