Searched refs:pre_split_size (Results 1 – 1 of 1) sorted by relevance
227 const int64 pre_split_size = original_shape.dimensions(shape_pair.split_dim); in CombineShapePair() local231 RoundUpToNearest<int64>(pre_split_size, 2); in CombineShapePair()233 if (reshape_dims[shape_pair.split_dim] != pre_split_size) { in CombineShapePair()