Searched refs:old_space_dims (Results 1 – 1 of 1) sorted by relevance
195 absl::Span<const int64_t> old_space_dims);2290 absl::Span<const int64_t> old_space_dims) { in SelectValidPortion() argument2295 << old_batch_dim << " old_space_dim " << old_space_dims[0]; in SelectValidPortion()2299 const int64_t old_space_size = old_shape.dimensions(old_space_dims[0]); in SelectValidPortion()