Searched refs:_global_batch_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ | ||
| D | distributed_training_utils.py | 29 return distribution_strategy.extended._global_batch_size # pylint: disable=protected-access |
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | one_device_strategy.py | 466 def _global_batch_size(self): member in OneDeviceExtended |
| D | parameter_server_strategy.py | 691 def _global_batch_size(self): member in ParameterServerStrategyExtended |
| D | collective_all_reduce_strategy.py | 920 def _global_batch_size(self): member in CollectiveAllReduceExtended |
| D | mirrored_strategy.py | 843 def _global_batch_size(self): member in MirroredExtended |
| D | strategy_test_lib.py | 440 if not strategy.extended._global_batch_size: # pylint: disable=protected-access |
| D | tpu_strategy.py | 1418 def _global_batch_size(self): member in TPUExtended |
| D | distribute_lib.py | 3572 def _global_batch_size(self): member in _DefaultDistributionExtended |