Searched refs:flatIndex (Results 1 – 3 of 3) sorted by relevance
62 uint32_t* flatIndex) const { in indexToFlatIndex()65 *flatIndex = 0; in indexToFlatIndex()67 *flatIndex += strides[i] * index[i]; in indexToFlatIndex()73 uint32_t* flatIndex) const { in broadcastedIndexToFlatIndex()76 *flatIndex = 0; in broadcastedIndexToFlatIndex()82 *flatIndex += strides[strides.size() - i] * index[index.size() - i]; in broadcastedIndexToFlatIndex()
45 bool indexToFlatIndex(const std::vector<uint32_t>& index, uint32_t* flatIndex) const;55 bool broadcastedIndexToFlatIndex(const std::vector<uint32_t>& index, uint32_t* flatIndex) const;
160 int flatIndex = 0; in getFlatIndex() local169 flatIndex += indexInInput * evalData.inputDimsNumElements[i]; in getFlatIndex()172 return flatIndex; in getFlatIndex()