Helper to reverse the order of dimension lengths in the passed-in literal.
| 939 | |
| 940 | // Helper to reverse the order of dimension lengths in the passed-in literal. |
| 941 | std::vector<int64> GetDimensionLengths(const Literal& literal) { |
| 942 | auto dimensions = literal.shape().dimensions(); |
| 943 | return std::vector<int64>(dimensions.rbegin(), dimensions.rend()); |
| 944 | } |
| 945 | |
| 946 | // Helper to compute strides for creating linear indices into multidimensional |
| 947 | // data from the dimension lengths and the layout. Returns a new vector of size |
no test coverage detected