Searched refs:max_shard_size (Results 1 – 1 of 1) sorted by relevance
637 int64 max_shard_size = shard_size_function.MaxInRange(0, shard_count); in ReshardAsWindowedInput() local638 shard_shape.set_dimensions(i, max_shard_size); in ReshardAsWindowedInput()665 (max_shard_size - 1) * wd.base_dilation() + 1; in ReshardAsWindowedInput()