diff options
Diffstat (limited to 'src/armnn/Descriptors.cpp')
-rw-r--r-- | src/armnn/Descriptors.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/armnn/Descriptors.cpp b/src/armnn/Descriptors.cpp index ab68097247..ef55ee7bb5 100644 --- a/src/armnn/Descriptors.cpp +++ b/src/armnn/Descriptors.cpp @@ -441,4 +441,15 @@ uint32_t FullyConnectedDescriptor::GetNumInputs() const return numInputs; } +uint32_t Convolution3dDescriptor::GetNumInputs() const +{ + // Return 2 otherwise check if bias is enabled + unsigned int numInputs = 2; + if (m_BiasEnabled) + { + numInputs = 3; + } + return numInputs; +} + } |