Searched refs:data_partition_id (Results 1 – 1 of 1) sorted by relevance
779 auto gen_slice = [&](HloInstruction* data_partition_id, in PartitionBaseCase()829 state.partition_id = data_partition_id; in PartitionBaseCase()1060 auto data_partition_id = in PartitionBaseCase() local1066 data_partition_id = body_b.AddInstruction( in PartitionBaseCase()1068 data_partition_id, partition_count)); in PartitionBaseCase()1081 state.partition_id = data_partition_id; in PartitionBaseCase()1119 data_partition_id, &body_b); in PartitionBaseCase()