Home
last modified time | relevance | path

Searched refs:CollectiveAllReduceExtended (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/integration_test/
Dmwms_peer_failure_test.py37 mwms_lib.CollectiveAllReduceExtended._enable_check_health = True
38 mwms_lib.CollectiveAllReduceExtended._check_health_interval = 3
39 mwms_lib.CollectiveAllReduceExtended._check_health_initial_timeout = 0
41 mwms_lib.CollectiveAllReduceExtended._check_health_timeout = 1
189 mwms_lib.CollectiveAllReduceExtended._check_alive_interval = 30
190 mwms_lib.CollectiveAllReduceExtended._check_alive_initial_timeout = 30
/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_continuous_run_test.py47 CollectiveAllReduceExtended = ( variable
48 collective_all_reduce_strategy.CollectiveAllReduceExtended)
49 CollectiveAllReduceExtended._enable_check_health = False
Dcollective_all_reduce_strategy.py186 CollectiveAllReduceExtended(
286 CollectiveAllReduceExtended(
299 class CollectiveAllReduceExtended(mirrored_strategy.MirroredExtended): class
383 super(CollectiveAllReduceExtended, self)._initialize_single_worker(
498 super(CollectiveAllReduceExtended, self)._initialize_single_worker(
579 return super(CollectiveAllReduceExtended,
Dstrategy_combinations.py45 CollectiveAllReduceExtended = ( variable
46 collective_all_reduce_strategy.CollectiveAllReduceExtended)
149 CollectiveAllReduceExtended._enable_check_health = False # pylint: disable=protected-access
Dtest_util.py60 collective_all_reduce_strategy.CollectiveAllReduceExtended):
Dcollective_all_reduce_strategy_test.py63 CollectiveAllReduceExtended = ( variable
64 collective_all_reduce_strategy.CollectiveAllReduceExtended)
Dvalues_test.py874 collective_all_reduce_strategy.CollectiveAllReduceExtended)