diff options
Diffstat (limited to 'src/armnn/Descriptors.cpp')
| -rw-r--r-- | src/armnn/Descriptors.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Descriptors.cpp b/src/armnn/Descriptors.cpp index 4521894c28..ab68097247 100644 --- a/src/armnn/Descriptors.cpp +++ b/src/armnn/Descriptors.cpp @@ -425,6 +425,11 @@ int StridedSliceDescriptor::GetStopForAxis(const TensorShape& inputShape, } +uint32_t FullyConnectedDescriptor::GetNumViews() const +{ + return GetNumInputs(); +} + uint32_t FullyConnectedDescriptor::GetNumInputs() const { // Return 2 otherwise check if bias is enabled |