diff options
Diffstat (limited to 'src/armnn/Network.hpp')
| -rw-r--r-- | src/armnn/Network.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index 8f16be1684..30941ca9e4 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -104,6 +104,11 @@ public: const char* name = nullptr); IConnectableLayer* AddFullyConnectedLayer(const FullyConnectedDescriptor& fullyConnectedDescriptor, + const Optional<ConstTensor>& weights, + const Optional<ConstTensor>& biases, + const char* name = nullptr); + + IConnectableLayer* AddFullyConnectedLayer(const FullyConnectedDescriptor& fullyConnectedDescriptor, const ConstTensor& weights, const Optional<ConstTensor>& biases, const char* name = nullptr); @@ -265,7 +270,7 @@ public: private: IConnectableLayer* AddFullyConnectedLayerImpl(const FullyConnectedDescriptor& fullyConnectedDescriptor, - const ConstTensor& weights, + const Optional<ConstTensor>& weights, const Optional<ConstTensor>& biases, const char* name); |