Searched refs:slice_sharding (Results 1 – 1 of 1) sorted by relevance
703 const HloSharding* slice_sharding; in PartitionBaseCase() local705 slice_sharding = windowed_op_is_lhs in PartitionBaseCase()709 slice_sharding = windowed_op_is_lhs in PartitionBaseCase()713 slice_sharding = windowed_op_is_lhs in PartitionBaseCase()717 CHECK_EQ(Product(slice_sharding->tile_assignment().dimensions()), in PartitionBaseCase()720 for (int64 i = 0; i < slice_sharding->tile_assignment().num_dimensions(); in PartitionBaseCase()722 if (slice_sharding->tile_assignment().dim(i) > 1) { in PartitionBaseCase()774 slice_operand, *slice_sharding, &body_b); in PartitionBaseCase()833 *slice_sharding); in PartitionBaseCase()1083 const HloSharding* slice_sharding; in PartitionBaseCase() local[all …]