diff options
Diffstat (limited to 'src/armnn/Descriptors.cpp')
-rw-r--r-- | src/armnn/Descriptors.cpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnn/Descriptors.cpp b/src/armnn/Descriptors.cpp index 881023e968..706992ccb0 100644 --- a/src/armnn/Descriptors.cpp +++ b/src/armnn/Descriptors.cpp @@ -425,4 +425,21 @@ int StridedSliceDescriptor::GetStopForAxis(const TensorShape& inputShape, } +uint32_t FullyConnectedDescriptor::GetNumViews() const +{ + // Return 1 with constant weights, otherwise check if bias is enabled + uint32_t numInputs = 1; + if (!m_ConstantWeights) + { + // non-const weights + numInputs = 2; + if (m_BiasEnabled) + { + // non-const bias + numInputs = 3; + } + } + return numInputs; +} + } |