| 822 | |
| 823 | template <typename T, size_t NDIMS> |
| 824 | typename TTypes<T, NDIMS>::Tensor Tensor::shaped( |
| 825 | gtl::ArraySlice<int64> new_sizes) { |
| 826 | CheckTypeAndIsAligned(DataTypeToEnum<T>::v()); |
| 827 | Eigen::array<Eigen::DenseIndex, NDIMS> dims; |
| 828 | FillDimsAndValidateCompatibleShape(new_sizes, &dims); |
| 829 | return typename TTypes<T, NDIMS>::Tensor(base<T>(), dims); |
| 830 | } |
| 831 | |
| 832 | template <typename T, size_t NDIMS> |
| 833 | typename TTypes<T, NDIMS>::Tensor Tensor::bit_casted_shaped( |