Searched refs:PerAxisIterator (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | BaseIterator.hpp | 694 class PerAxisIterator : public Base class
697 PerAxisIterator(T* data = nullptr, in PerAxisIterator() function in armnn::PerAxisIterator
708 PerAxisIterator(T* data = nullptr, in PerAxisIterator() function in armnn::PerAxisIterator
728 PerAxisIterator& operator++() override in operator ++()
735 PerAxisIterator& operator+=(const unsigned int increment) override in operator +=()
742 PerAxisIterator& operator-=(const unsigned int decrement) override in operator -=()
750 inline PerAxisIterator& SetIndexOnMem(const unsigned int index) in SetIndexOnMem()
766 PerAxisIterator& operator[](const unsigned int index) override in operator []()
781 class QSymm8PerAxisDecoder : public PerAxisIterator<const int8_t, Decoder<float>>
785 : PerAxisIterator(data, tensorInfo.GetShape(), tensorInfo.GetQuantizationDim().value()), in QSymm8PerAxisDecoder()
[all …]
|
| /external/armnn/src/backends/reference/test/ |
| D | RefPerAxisIteratorTests.cpp | 42 class MockPerAxisIterator : public PerAxisIterator<const int8_t, Decoder<int8_t>>
46 : PerAxisIterator(data, tensorShape, axis), m_NumElements(tensorShape.GetNumElements()) in MockPerAxisIterator()
|