Searched refs:output_shard_size (Results 1 – 1 of 1) sorted by relevance
2051 int64 output_shard_size = output_shard_shape.dimensions(output_sharded_dim); in HandleReshape() local2063 dim->set_padding_high(output_shard_size * split_factor * in HandleReshape()2081 output_shard_size * split_factor); in HandleReshape()