Searched defs:StridedData (Results 1 – 2 of 2) sorted by relevance
15 struct StridedData { struct16 StridedData(const Tensor& tensor) in StridedData() argument21 void* data;22 IntArrayRef strides;23 int64_t elementSize;25 void step(int dim) { in step()36 std::array<StridedData, N> strided_data) { in recursive_apply()
90 struct StridedData { struct91 typedef impl::Entry<T> Entry;97 __device__ int get_index(int i) const { return (*this)[i].index; } in get_index()98 __device__ T get_value(int i) const { return (*this)[i].value; } in get_value()100 Entry* const data;