Searched refs:windowed_op_is_lhs (Results 1 – 1 of 1) sorted by relevance
605 const bool windowed_op_is_lhs = in PartitionBaseCase() local614 windowed_op_is_lhs ? *lhs_sharding_transposed_to_match_output in PartitionBaseCase()620 auto& to_mask = windowed_op_is_lhs ? lhs : rhs; in PartitionBaseCase()636 : windowed_op_is_lhs ? lhs.hlo() in PartitionBaseCase()705 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()730 if (windowed_op_is_lhs) { in PartitionBaseCase()737 windowed_op_is_lhs in PartitionBaseCase()741 windowed_op_is_lhs in PartitionBaseCase()[all …]