Home | History | Annotate | Download | only in Tensor

Lines Matching refs:array_get

47     return array_get<RowMajor ? n - 1 : (NumIndices - n)>(indices) +
71 return array_get<n-1>(dimensions) * mult +
240 return array_get<RowMajor ? n : (NumIndices - n - 1)>(indices) +
241 array_get<RowMajor ? n : (NumIndices - n - 1)>(dimensions) *
252 return array_get<RowMajor ? 0 : NumIndices - 1>(indices);
344 return array_get<RowMajor ? n : (NumIndices - n - 1)>(indices) +
345 array_get<RowMajor ? n : (NumIndices - n - 1)>(dimensions) *
356 return array_get<RowMajor ? 0 : NumIndices - 1>(indices);
377 template <std::ptrdiff_t n, typename std::ptrdiff_t... Indices> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::ptrdiff_t array_get(const Sizes<Indices...>&) {
380 template <std::ptrdiff_t n> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::ptrdiff_t array_get(const Sizes<>&) {
391 template <std::size_t n, std::size_t V1, std::size_t V2, std::size_t V3, std::size_t V4, std::size_t V5> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::size_t array_get(const Sizes<V1,V2,V3,V4,V5>&) {
407 return (array_get<n-1>(dims1) == array_get<n-1>(dims2)) &