diff options
Diffstat (limited to 'src/armnn/InternalTypes.cpp')
-rw-r--r-- | src/armnn/InternalTypes.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/armnn/InternalTypes.cpp b/src/armnn/InternalTypes.cpp index fce1e95429..67c596eec7 100644 --- a/src/armnn/InternalTypes.cpp +++ b/src/armnn/InternalTypes.cpp @@ -48,4 +48,16 @@ char const* GetLayerTypeAsCString(LayerType type) } } +// Definition in include/armnn/Types.hpp +bool operator==(const DataLayout& dataLayout, const DataLayoutIndexed& indexed) +{ + return dataLayout == indexed.GetDataLayout(); +} + +// Definition in include/armnn/Types.hpp +bool operator==(const DataLayoutIndexed& indexed, const DataLayout& dataLayout) +{ + return indexed.GetDataLayout() == dataLayout; +} + } |