Searched defs:IndexToOffset (Results 1 – 3 of 3) sorted by relevance
77 struct IndexToOffset { struct78 static __host__ __device__ IndexType get( in get()98 struct IndexToOffset<T, IndexType, -1> { struct99 static inline __host__ __device__ IndexType get( in get()
36 struct IndexToOffset { struct37 const IntArrayRef sizes;38 const IntArrayRef strides;39 const int64_t ndim;40 explicit IndexToOffset(const TensorBase & tensor) : in IndexToOffset() argument44 int64_t get(int64_t linear_index) const { in get()
1438 static int64_t IndexToOffset(ArrayRef<int64_t> shape, in IndexToOffset() function in mlir::TF::__anon09b86ffb0111::StridedArrayViewBase