diff options
Diffstat (limited to 'include/armnnUtils/DataLayoutIndexed.hpp')
-rw-r--r-- | include/armnnUtils/DataLayoutIndexed.hpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/include/armnnUtils/DataLayoutIndexed.hpp b/include/armnnUtils/DataLayoutIndexed.hpp index 03404bda5d..c6701f7d5c 100644 --- a/include/armnnUtils/DataLayoutIndexed.hpp +++ b/include/armnnUtils/DataLayoutIndexed.hpp @@ -13,7 +13,7 @@ namespace armnnUtils { -// Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout +/// Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout class DataLayoutIndexed { public: @@ -36,25 +36,25 @@ public: BOOST_ASSERT( widthIndex < shape[m_WidthIndex] || ( shape[m_WidthIndex] == 0 && widthIndex == 0) ); - // Offset the given indices appropriately depending on the data layout + /// Offset the given indices appropriately depending on the data layout switch (m_DataLayout) { case armnn::DataLayout::NHWC: batchIndex *= shape[1] * shape[2] * shape[3]; // batchIndex *= heightIndex * widthIndex * channelIndex heightIndex *= shape[m_WidthIndex] * shape[m_ChannelsIndex]; widthIndex *= shape[m_ChannelsIndex]; - // channelIndex stays unchanged + /// channelIndex stays unchanged break; case armnn::DataLayout::NCHW: default: batchIndex *= shape[1] * shape[2] * shape[3]; // batchIndex *= heightIndex * widthIndex * channelIndex channelIndex *= shape[m_HeightIndex] * shape[m_WidthIndex]; heightIndex *= shape[m_WidthIndex]; - // widthIndex stays unchanged + /// widthIndex stays unchanged break; } - // Get the value using the correct offset + /// Get the value using the correct offset return batchIndex + channelIndex + heightIndex + widthIndex; } @@ -65,7 +65,7 @@ private: unsigned int m_WidthIndex; }; -// Equality methods +/// Equality methods bool operator==(const armnn::DataLayout& dataLayout, const DataLayoutIndexed& indexed); bool operator==(const DataLayoutIndexed& indexed, const armnn::DataLayout& dataLayout); |