90     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 0);  variable
91     std::vector<int8_t> output = iterator.Loop();
95     iterator[5];
96     CHECK(iterator.GetAxisIndex() == 1u);
98     iterator[1];
99     CHECK(iterator.GetAxisIndex() == 0u);
101     iterator[10];
102     CHECK(iterator.GetAxisIndex() == 2u);
113     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 1);  variable
114     std::vector<int8_t> output = iterator.Loop();
118     iterator[5];
119     CHECK(iterator.GetAxisIndex() == 0u);
121     iterator[1];
122     CHECK(iterator.GetAxisIndex() == 0u);
124     iterator[10];
125     CHECK(iterator.GetAxisIndex() == 0u);
136     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 2);  variable
137     std::vector<int8_t> output = iterator.Loop();
141     iterator[5];
142     CHECK(iterator.GetAxisIndex() == 0u);
144     iterator[1];
145     CHECK(iterator.GetAxisIndex() == 0u);
147     iterator[10];
148     CHECK(iterator.GetAxisIndex() == 1u);
159     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 3);  variable
160     std::vector<int8_t> output = iterator.Loop();
164     iterator[5];
165     CHECK(iterator.GetAxisIndex() == 1u);
167     iterator[1];
168     CHECK(iterator.GetAxisIndex() == 1u);
170     iterator[10];
171     CHECK(iterator.GetAxisIndex() == 0u);
195     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 1);  variable
196     std::vector<int8_t> output = iterator.Loop();
200     iterator[5];
201     CHECK(iterator.GetAxisIndex() == 1u);
203     iterator[1];
204     CHECK(iterator.GetAxisIndex() == 0u);
206     iterator[10];
207     CHECK(iterator.GetAxisIndex() == 0u);
229     auto iterator = MockPerAxisIterator(input.data(), tensorInfo.GetShape(), 2);  variable
231     iterator += 3;
232     CHECK(iterator.Get() == expOutput[3]);
233     CHECK(iterator.GetAxisIndex() == 1u);
235     iterator += 3;
236     CHECK(iterator.Get() == expOutput[6]);
237     CHECK(iterator.GetAxisIndex() == 1u);
239     iterator -= 2;
240     CHECK(iterator.Get() == expOutput[4]);
241     CHECK(iterator.GetAxisIndex() == 0u);
243     iterator -= 1;
244     CHECK(iterator.Get() == expOutput[3]);
245     CHECK(iterator.GetAxisIndex() == 1u);