Searched refs:_call_for_each_replica (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | mirrored_strategy.py | 27 _call_for_each_replica = mirrored_strategy._call_for_each_replica variable
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 363 def _call_for_each_replica(self, fn, args, kwargs): member in ParameterServerStrategyExtended
365 return mirrored_strategy._call_for_each_replica(
|
| D | distribute_lib.py | 1128 return self._call_for_each_replica(fn, args, kwargs)
1130 def _call_for_each_replica(self, fn, args, kwargs): member in DistributionStrategyExtended
1599 def _call_for_each_replica(self, fn, args, kwargs): member in _DefaultDistributionExtended
|
| D | distribute_lib_test.py | 51 def _call_for_each_replica(self, fn, args, kwargs): member in _TestExtended
|
| D | mirrored_strategy.py | 86 def _call_for_each_replica(distribution, device_map, fn, args, kwargs): function
663 def _call_for_each_replica(self, fn, args, kwargs): member in MirroredExtended
664 return _call_for_each_replica(self._container_strategy(), self._device_map,
|
| D | one_device_strategy.py | 141 def _call_for_each_replica(self, fn, args, kwargs): member in OneDeviceExtended
|
| D | tpu_strategy.py | 383 def _call_for_each_replica(self, fn, args, kwargs): member in TPUExtended
|