| 754 | |
| 755 | template <typename T, size_t NDIMS> |
| 756 | typename TTypes<T, NDIMS>::Tensor Tensor::reinterpret_last_dimension() { |
| 757 | if (NDIMS == dims()) { |
| 758 | return tensor<T, NDIMS>(); |
| 759 | } |
| 760 | CHECK(IsAligned()); |
| 761 | CHECK_EQ(NDIMS, dims() - 1); |
| 762 | CHECK_EQ(sizeof(T), shape_.dim_sizes()[NDIMS] * DataTypeSize(dtype())); |
| 763 | Eigen::array<Eigen::DenseIndex, NDIMS> dims; |
| 764 | for (int d = 0; d < NDIMS; ++d) { |
| 765 | dims[d] = shape_.dim_sizes()[d]; |
| 766 | } |
| 767 | return typename TTypes<T, NDIMS>::Tensor(base<T>(), dims); |
| 768 | } |
| 769 | |
| 770 | template <typename T, size_t NDIMS> |
| 771 | typename TTypes<T, NDIMS>::ConstTensor Tensor::reinterpret_last_dimension() |
nothing calls this directly
no test coverage detected