Searched refs:num_replicas_in_graph (Results 1 – 4 of 4) sorted by relevance
67 def num_replicas_in_graph(self): member in DeviceMap127 def num_replicas_in_graph(self): member in SingleDeviceMap188 def num_replicas_in_graph(self): member in ReplicaDeviceMap1303 assert len(values) == device_map.num_replicas_in_graph
92 if device_map.num_replicas_in_graph != 1:453 num_replicas = device_map.num_replicas_in_graph * num_between_graph_workers
116 for index in range(device_map.num_replicas_in_graph):754 return self._device_map.num_replicas_in_graph
514 return self._device_map.num_replicas_in_graph