Searched refs:cp_b (Results 1 – 1 of 1) sorted by relevance
1349 SpmdBuilder cp_b("window_collective_permute", original_hlo); in PartitionBaseCase() local1351 auto p = cp_b.AddInstruction(HloInstruction::CreateParameter( in PartitionBaseCase()1365 &cp_b, p, sd_pairs, (*lhs.state().next_channel_id)++); in PartitionBaseCase()1384 module->AddEmbeddedComputation(cp_b.Build()), in PartitionBaseCase()