diff options
| author | Rob Hughes <robert.hughes@arm.com> | 2018-11-21 13:34:24 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-11-26 13:26:07 +0000 |
| commit | 144c01b56d5e8b2f9d8e84a03f7fe975888ee25a (patch) | |
| tree | c3e31a087d26b955cad26cb41de51d3f5ba45b6d /include | |
| parent | 0214d7e7d075ae2b8534847d4d5fd9c485626694 (diff) | |
| download | armnn-144c01b56d5e8b2f9d8e84a03f7fe975888ee25a.tar.gz | |
Fix documentation for depthwise convolution
Change-Id: Ib2bf543fc4f855da94eb1d34a91637f7868321cd
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/INetwork.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index ac7faab279..ab50abcd4d 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -112,7 +112,7 @@ public: /// Adds a 2D depthwise convolution layer to the network. /// @param convolution2dDescriptor - Description of the 2D depthwise convolution layer. - /// @param weights - Tensor for the weights data. Expected format: [1, outputChannels, height, width]. + /// @param weights - Tensor for the weights. Expected format: [channelMultiplier, inputChannels, height, width]. /// @param biases (Optional) - Tensor for the bias data. Must match the output tensor shape. /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. |