Searched refs:_make_per_replica (Results 1 – 1 of 1) sorted by relevance
63 def _make_per_replica(values, devices, regroup=False): function168 per_replica = _make_per_replica(values, devices)172 per_replica_2 = _make_per_replica(values_2, devices)575 per_replica = _make_per_replica(values, devices)579 per_replica_2 = _make_per_replica(values_2, devices)814 v0 = _make_per_replica([1.0 for _ in devices], devices)815 v1 = _make_per_replica([2.0 for _ in devices], devices)