Searched defs:_call_for_each_replica (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_run.py | 125 def _call_for_each_replica(distribution, fn, args, kwargs): function
|
| D | parameter_server_strategy_v2.py | 718 def _call_for_each_replica(self, fn, args, kwargs): member in ParameterServerStrategyV2Extended
|
| D | one_device_strategy.py | 391 def _call_for_each_replica(self, fn, args, kwargs): member in OneDeviceExtended
|
| D | parameter_server_strategy.py | 488 def _call_for_each_replica(self, fn, args, kwargs): member in ParameterServerStrategyExtended
|
| D | mirrored_strategy.py | 669 def _call_for_each_replica(self, fn, args, kwargs): member in MirroredExtended
|
| D | distribute_lib_test.py | 74 def _call_for_each_replica(self, fn, args, kwargs): member in _TestExtended
|
| D | tpu_strategy.py | 1086 def _call_for_each_replica(self, fn, args, kwargs): member in TPUExtended
|
| D | distribute_lib.py | 2776 def _call_for_each_replica(self, fn, args, kwargs): member in StrategyExtendedV1
3461 def _call_for_each_replica(self, fn, args, kwargs): member in _DefaultDistributionExtended
|