Searched defs:OffsetCalculator (Results 1 – 1 of 1) sorted by relevance
23 struct OffsetCalculator { struct36 …OffsetCalculator(int dims, const int64_t* sizes, const int64_t* const* strides, const int64_t* ele… in OffsetCalculator() argument47 C10_HOST_DEVICE offset_type get(index_t linear_idx) const { in get()77 struct TrivialOffsetCalculator { argument95 // Make an OffsetCalculator with byte offsets argument