Searched refs:_container_strategy (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 294 self._container_strategy())
302 self._container_strategy())
325 self._container_strategy())
339 self._container_strategy())
392 strategy = self._container_strategy()
|
| D | parameter_server_strategy.py | 345 self._container_strategy(),
352 self._container_strategy(),
374 self._container_strategy())
399 self._container_strategy())
449 wrapped = ps_values.AggregatingVariable(self._container_strategy(), v,
489 return mirrored_run.call_for_each_replica(self._container_strategy(), fn,
|
| D | mirrored_strategy.py | 529 self._container_strategy(), _real_mirrored_creator,
541 self._container_strategy(),
557 self._container_strategy())
570 self._container_strategy(),
588 dataset_fn, input_workers, input_contexts, self._container_strategy(),
671 self._container_strategy(), fn, args, kwargs)
|
| D | distribute_lib.py | 306 strategy = extended._container_strategy() # pylint: disable=protected-access
356 strategy = extended._container_strategy() # pylint: disable=protected-access
2054 def _container_strategy(self): member in StrategyExtendedV2
2537 with self._container_strategy().scope():
2718 with self._container_strategy().scope():
2773 with self._container_strategy().scope():
2818 with self._container_strategy().scope():
3462 with ReplicaContext(self._container_strategy(), replica_id_in_sync_group=0):
|
| D | tpu_strategy.py | 884 self._container_strategy(),
904 self._container_strategy())
948 self._container_strategy(),
972 self._container_strategy())
1089 with _TPUReplicaContext(self._container_strategy()):
1164 self._container_strategy(), _real_mirrored_creator,
1436 strategy = self._container_strategy()
|
| D | collective_all_reduce_strategy.py | 605 self._container_strategy(),
622 strategy=self._container_strategy())
641 self._container_strategy(),
653 self._container_strategy())
|
| D | parameter_server_strategy_v2.py | 721 self._container_strategy(),
|
| D | distribute_lib_test.py | 76 self._container_strategy(), replica_id_in_sync_group=0):
89 self._container_strategy())
|