Searched refs:_OneDeviceReplicaContext (Results 1 – 1 of 1) sorted by relevance
393 with ops.device(self._device), _OneDeviceReplicaContext(strategy):478 class _OneDeviceReplicaContext(distribute_lib.ReplicaContext): class