Lines Matching refs:i
31 for (int i = strides.size() - 2; i >= 0; --i) { in IndexedShapeWrapper() local
32 strides[i] = shape->dimensions[i + 1] * strides[i + 1]; in IndexedShapeWrapper()
40 for (size_t i = 0; i < index->size(); ++i) { in nextIndexInplace() local
41 if (index->at(i) < shape->dimensions[i] - 1) { in nextIndexInplace()
50 for (int i = index->size() - 1; i >= 0; --i) { in nextIndexInplace() local
51 ++index->at(i); in nextIndexInplace()
52 if (index->at(i) == shape->dimensions[i]) { in nextIndexInplace()
53 index->at(i) = 0; in nextIndexInplace()
66 for (size_t i = 0; i < index.size(); ++i) { in indexToFlatIndex() local
67 *flatIndex += strides[i] * index[i]; in indexToFlatIndex()
77 for (size_t i = 1; i <= strides.size(); ++i) { in broadcastedIndexToFlatIndex() local
78 uint32_t currentIndex = index[index.size() - i]; in broadcastedIndexToFlatIndex()
79 uint32_t currentDimSize = shape->dimensions[shape->dimensions.size() - i]; in broadcastedIndexToFlatIndex()
82 *flatIndex += strides[strides.size() - i] * index[index.size() - i]; in broadcastedIndexToFlatIndex()
95 for (size_t i = 0; i < index.size(); ++i) { in isValid() local
96 if (index[i] >= shape->dimensions[i]) { in isValid()