Searched refs:same_group_assignments (Results 1 – 2 of 2) sorted by relevance
41 bool same_group_assignments(mlir::DenseIntElementsAttr attr_a, in same_group_assignments() function125 same_group_assignments(all_reduce_group_assignment_attr, in ApplyOptimization()
487 bool same_group_assignments(mlir::Value group_assignment_a, in same_group_assignments() function522 if (same_group_assignments(group_assignment, seen_group_assignment)) { in CombineAllReduceOpsOfSameType()