Searched refs:CollectiveAllReduceStrategy (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 31 from tensorflow.python.distribute.collective_all_reduce_strategy import CollectiveAllReduceStrategy
171 if isinstance(strategy, CollectiveAllReduceStrategy
192 if isinstance(strategy, CollectiveAllReduceStrategy):
228 if isinstance(strategy, CollectiveAllReduceStrategy
244 if isinstance(strategy, CollectiveAllReduceStrategy):
488 if isinstance(strategy, CollectiveAllReduceStrategy
510 if isinstance(strategy, CollectiveAllReduceStrategy):
550 if isinstance(strategy, CollectiveAllReduceStrategy
567 if isinstance(strategy, CollectiveAllReduceStrategy):
667 if isinstance(strategy, CollectiveAllReduceStrategy):
|
| D | collective_all_reduce_strategy_test.py | 61 CollectiveAllReduceStrategy = ( variable
62 collective_all_reduce_strategy.CollectiveAllReduceStrategy)
89 strategy = collective_all_reduce_strategy.CollectiveAllReduceStrategy(
102 CollectiveAllReduceStrategy._collective_key_base += 100000
541 collective_all_reduce_strategy.CollectiveAllReduceStrategy(
579 strategy = CollectiveAllReduceStrategy()
581 self.assertIsInstance(strategy, CollectiveAllReduceStrategy)
583 self.assertIsInstance(experimental_strategy, CollectiveAllReduceStrategy)
589 self.assertEqual(CollectiveAllReduceStrategy.__name__,
|
| D | collective_all_reduce_strategy.py | 57 class CollectiveAllReduceStrategy(distribute_lib.Strategy): class
185 super(CollectiveAllReduceStrategy, self).__init__(
226 return isinstance(instance, CollectiveAllReduceStrategy)
231 CollectiveAllReduceStrategy,
234 __doc__ = CollectiveAllReduceStrategy.__doc__
268 _CollectiveAllReduceStrategyExperimental.__name__ = CollectiveAllReduceStrategy.__name__
274 __doc__ = CollectiveAllReduceStrategy.__doc__
|
| D | strategy_common_test.py | 32 from tensorflow.python.distribute.collective_all_reduce_strategy import CollectiveAllReduceStrategy
175 if isinstance(strategy, CollectiveAllReduceStrategy):
320 if not isinstance(strategy, CollectiveAllReduceStrategy) and not isinstance(
333 if isinstance(strategy, CollectiveAllReduceStrategy):
|
| D | multi_worker_continuous_run_test.py | 75 strategy = collective_all_reduce_strategy.CollectiveAllReduceStrategy()
108 strategy = collective_all_reduce_strategy.CollectiveAllReduceStrategy()
|
| D | strategy_combinations.py | 67 CollectiveAllReduceStrategy = ( variable
68 collective_all_reduce_strategy.CollectiveAllReduceStrategy)
154 strategy = CollectiveAllReduceStrategy(cluster_resolver=resolver)
|
| D | strategy_combinations_test.py | 124 strategy, collective_all_reduce_strategy.CollectiveAllReduceStrategy)
177 strategy, collective_all_reduce_strategy.CollectiveAllReduceStrategy)
|
| D | moving_averages_test.py | 197 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
260 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
|
| D | combinations.py | 576 strategy, collective_all_reduce_strategy.CollectiveAllReduceStrategy):
|
| D | mirrored_variable_test.py | 68 CollectiveAllReduceStrategy._from_local_devices((
|
| D | values_test.py | 431 collective_all_reduce_strategy.CollectiveAllReduceStrategy)
835 collective_all_reduce_strategy.CollectiveAllReduceStrategy)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | mirrored_variable_test.py | 51 CollectiveAllReduceStrategy._from_local_devices((
|
| D | multi_worker_test.py | 191 collective_strategy.CollectiveAllReduceStrategy,
|
| D | multi_worker_callback_tf2_test.py | 70 with collective_strategy.CollectiveAllReduceStrategy().scope():
|
| D | distribute_strategy_test.py | 1853 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
1882 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
1909 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
1948 collective_all_reduce_strategy.CollectiveAllReduceStrategy):
|
| /external/tensorflow/tensorflow/python/profiler/integration_test/ |
| D | profiler_api_test.py | 48 with collective_strategy.CollectiveAllReduceStrategy().scope():
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 4 …class \'tensorflow.python.distribute.collective_all_reduce_strategy.CollectiveAllReduceStrategy\'>"
|
| D | tensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt | 3 …class \'tensorflow.python.distribute.collective_all_reduce_strategy.CollectiveAllReduceStrategy\'>"
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 454 mwms_lib.CollectiveAllReduceStrategy._collective_key_base = (
474 strategy = (mwms_lib.CollectiveAllReduceStrategy
505 strategy = mwms_lib.CollectiveAllReduceStrategy(
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 1183 collective_all_reduce_strategy.CollectiveAllReduceStrategy,
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 2797 collective_all_reduce_strategy.CollectiveAllReduceStrategy)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 1613 collective_all_reduce_strategy.CollectiveAllReduceStrategy,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 538 strategy = collective_all_reduce_strategy.CollectiveAllReduceStrategy()
|