Searched refs:_container_strategy (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 104 strategy = extended._container_strategy() # pylint: disable=protected-access
152 strategy = extended._container_strategy() # pylint: disable=protected-access
848 def _container_strategy(self): member in DistributionStrategyExtended
1072 with self._container_strategy().scope():
1127 with self._container_strategy().scope():
1226 with self._container_strategy().scope():
1250 with self._container_strategy().scope():
1601 self._container_strategy(),
|
| D | parameter_server_strategy.py | 325 self._container_strategy(), v, aggregation)
366 self._container_strategy(), self._device_map, fn, args, kwargs)
|
| D | distribute_lib_test.py | 53 self._container_strategy(),
|
| D | one_device_strategy.py | 142 strategy = self._container_strategy()
|
| D | tpu_strategy.py | 386 with _TPUReplicaContext(self._container_strategy()):
438 self._container_strategy(), device_map, logical_device,
|
| D | mirrored_strategy.py | 563 self._container_strategy(), device_map, logical_device,
664 return _call_for_each_replica(self._container_strategy(), self._device_map,
|
| D | collective_all_reduce_strategy.py | 328 self._container_strategy(), device_map, logical_device,
|