Searched refs:dsizes (Results 1 – 2 of 2) sorted by relevance
476 Eigen::DSizes<IndexType, NDIMS> dsizes; in AsEigenDSizesWithPadding() local478 dsizes[d] = static_cast<IndexType>(dim_size(d)); in AsEigenDSizesWithPadding()481 dsizes[d] = 1; in AsEigenDSizesWithPadding()483 return dsizes; in AsEigenDSizesWithPadding()
38 Eigen::DSizes<Eigen::DenseIndex, kTensorSliceMaxRank> dsizes = in GetEigenTensorMapFromTensorShape() local41 data, dsizes); in GetEigenTensorMapFromTensorShape()