Searched refs:make_replica_groups (Results 1 – 2 of 2) sorted by relevance
677 def make_replica_groups(replica_groups): function
679 ops.Constant(c, lhs), xla_client.make_replica_groups([[0]]))