Searched refs:ReplicaContext (Results 1 – 12 of 12) sorted by relevance
1 path: "tensorflow.distribute.ReplicaContext"3 is_instance: "<class \'tensorflow.python.distribute.distribute_lib.ReplicaContext\'>"
60 name: "ReplicaContext"
1 path: "tensorflow.distribute.ReplicaContext"
40 name: "ReplicaContext"
478 class _OneDeviceReplicaContext(distribute_lib.ReplicaContext):482 distribute_lib.ReplicaContext.__init__(
720 with distribute_lib.ReplicaContext(
3152 class ReplicaContext(ReplicaContextBase): class3462 with ReplicaContext(self._container_strategy(), replica_id_in_sync_group=0):3577 class _DefaultReplicaContext(ReplicaContext):
361 class _MirroredReplicaContext(distribute_lib.ReplicaContext):
1527 class _TPUReplicaContext(distribute_lib.ReplicaContext):1533 distribute_lib.ReplicaContext.__init__(
42 class _TestReplicaContext(distribute_lib.ReplicaContext):
913 with distribute_lib.ReplicaContext(distribution, i):920 with distribute_lib.ReplicaContext(distribution, i):
432 `MirroredStrategy.run` or `ReplicaContext.merge_call`. This is because a thread local448 `tf.distribute.StrategyExtended.batch_reduce_to`, `tf.distribute.ReplicaContext.all_reduce`587 * Adds `tf.distribute.Strategy.gather` and `tf.distribute.ReplicaContext.all_gather`1135 …* Add `tf.distribute.Strategy.gather` and `tf.distribute.ReplicaContext.all_gather` methods to gat…1351 * AutoGraph is now available in `ReplicaContext.merge_call`,