Searched refs:cw_dot (Results 1 – 1 of 1) sorted by relevance
1020 auto cw_dot = body_b.AddInstruction(HloInstruction::CreateSlice( in PartitionBaseCase() local1029 o->shape(), HloOpcode::kAdd, cw_cp_output, cw_dot)); in PartitionBaseCase()1044 o->shape(), o, cw_dot, cw_offsets)); in PartitionBaseCase()