Searched refs:ReplicaContext (Results 1 – 11 of 11) sorted by relevance
1 path: "tensorflow.distribute.ReplicaContext"3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContext\'>"
64 name: "ReplicaContext"
1 path: "tensorflow.distribute.ReplicaContext"
40 name: "ReplicaContext"
478 class _OneDeviceReplicaContext(distribute_lib.ReplicaContext):482 distribute_lib.ReplicaContext.__init__(
590 with distribute_lib.ReplicaContext(distribution, i):597 with distribute_lib.ReplicaContext(distribution, i):
3298 class ReplicaContext(ReplicaContextBase): class3691 with ReplicaContext(self._container_strategy(), replica_id_in_sync_group=0):3806 class _DefaultReplicaContext(ReplicaContext):
429 class _MirroredReplicaContext(distribute_lib.ReplicaContext):
1690 class _TPUReplicaContext(distribute_lib.ReplicaContext):1696 distribute_lib.ReplicaContext.__init__(
39 class _TestReplicaContext(distribute_lib.ReplicaContext):
4546 `MirroredStrategy.run` or `ReplicaContext.merge_call`. This is because a4569 `tf.distribute.ReplicaContext.all_reduce` to `options`.4738 `tf.distribute.ReplicaContext.all_gather` APIs to support gathering dense5424 `tf.distribute.ReplicaContext.all_gather` methods to gather and concatenate5855 * AutoGraph is now available in `ReplicaContext.merge_call`,