Searched refs:_call_for_each_replica (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_run.py | 69 return _call_for_each_replica(strategy, fn, args, kwargs)
101 return _call_for_each_replica(strategy, fn, args, kwargs)
150 def _call_for_each_replica(distribution, fn, args, kwargs): function
|
| D | one_device_strategy.py | 391 def _call_for_each_replica(self, fn, args, kwargs): member in OneDeviceExtended
|
| D | distribute_lib.py | 2891 return self._call_for_each_replica(fn, args, kwargs)
2893 def _call_for_each_replica(self, fn, args, kwargs): member in StrategyExtendedV1
3690 def _call_for_each_replica(self, fn, args, kwargs): member in _DefaultDistributionExtended
|
| D | parameter_server_strategy_v2.py | 915 def _call_for_each_replica(self, fn, args, kwargs): member in ParameterServerStrategyV2Extended
|
| D | parameter_server_strategy.py | 491 def _call_for_each_replica(self, fn, args, kwargs): member in ParameterServerStrategyExtended
|
| D | distribute_lib_test.py | 71 def _call_for_each_replica(self, fn, args, kwargs): member in _TestExtended
|
| D | mirrored_strategy.py | 675 def _call_for_each_replica(self, fn, args, kwargs): member in MirroredExtended
|
| D | tpu_strategy.py | 1144 def _call_for_each_replica(self, fn, args, kwargs): member in TPUExtended
|