| 141 | // An index type with compile-time known size. |
| 142 | template <int IndexCount> |
| 143 | struct Index : Array<int, IndexCount, 0> { |
| 144 | typedef Array<int, IndexCount, 0> Base; |
| 145 | EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index() : Base() {} |
| 146 | EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0) : Base(a0) {} |
| 147 | EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0, int a1) : Base(a0, a1) {} |
| 148 | EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0, int a1, int a2) |
| 149 | : Base(a0, a1, a2) {} |
| 150 | }; |
| 151 | |
| 152 | // A helper function that converts a tensor index into a flat array index. |
| 153 | template <int IndexCount> |
no outgoing calls
no test coverage detected