aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/ISubGraphConverter.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/ISubGraphConverter.hpp')
-rw-r--r--src/armnn/ISubGraphConverter.hpp12
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