diff options
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. |