diff options
| author | Cathal Corbett <cathal.corbett@arm.com> | 2021-12-13 13:03:22 +0000 |
|---|---|---|
| committer | Cathal Corbett <cathal.corbett@arm.com> | 2021-12-14 14:09:55 +0000 |
| commit | 18655b8d326f6109c6fedacf42b46dc4bc942324 (patch) | |
| tree | 79a4cff6e46ea41f404a693ac3a9de2787498f06 /src/armnn/Network.hpp | |
| parent | a097d2a0ed8e30d5aaf6d29ec18d0c39201b7b67 (diff) | |
| download | armnn-18655b8d326f6109c6fedacf42b46dc4bc942324.tar.gz | |
IVGCVSW-6630 Add new method AddPrecompiledLayer() to INetwork
* Add new method AddPrecompiledLayer() to INetwork
with Comments noting it is for backend users.
* Added unit test to SubgraphViewTests.cpp.
* Bug fix and code refactor in Graph.cpp specifically around
Graph::SubstituteSubgraph(SubgraphView& subgraph,
IConnectableLayer* substituteLayer)
Change-Id: If2d816e5109e48ce920bf92d8823b39130c23a16
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Diffstat (limited to 'src/armnn/Network.hpp')
| -rw-r--r-- | src/armnn/Network.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index 959d88dbed..3fdc140099 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -170,6 +170,10 @@ public: IConnectableLayer* AddPooling3dLayer(const Pooling3dDescriptor& pooling3dDescriptor, const char* name = nullptr); + IConnectableLayer* AddPrecompiledLayer(const PreCompiledDescriptor& preCompiledDescriptor, + CompiledBlobPtr& compiledBlobPtr, + const Optional<BackendId>& backend); + IConnectableLayer* AddPreluLayer(const char* name = nullptr); IConnectableLayer* AddQuantizeLayer(const char* name = nullptr); |