diff options
Diffstat (limited to 'include/armnn/INetwork.hpp')
| -rw-r--r-- | include/armnn/INetwork.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 073f119ef4..6a2193cbc6 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -13,6 +13,7 @@ #include <armnn/Optional.hpp> #include <armnn/TensorFwd.hpp> #include <armnn/Logging.hpp> +#include <armnn/backends/TensorHandle.hpp> #include <memory> #include <vector> @@ -119,6 +120,11 @@ public: /// the BaseDescriptor IsNull function is invoked. virtual const BaseDescriptor& GetParameters() const = 0; + using ConstantTensors = std::vector<std::reference_wrapper<std::shared_ptr<ConstTensorHandle>>>; + + // Returns ConstantTensors of this Layer if it has any, otherwise returns empty vector. + virtual ConstantTensors GetConstantTensorsByRef() = 0; + protected: /// Objects are not deletable via the handle ~IConnectableLayer() {} |