Home
last modified time | relevance | path

Searched refs:mwms_lib (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/integration_test/
Dmwms_peer_failure_test.py29 from tensorflow.python.distribute import collective_all_reduce_strategy as mwms_lib unknown
37 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/v1/
Dcross_device_ops_test.py30 from tensorflow.python.distribute import collective_all_reduce_strategy as mwms_lib unknown
454 mwms_lib.CollectiveAllReduceStrategy._collective_key_base = (
474 strategy = (mwms_lib.CollectiveAllReduceStrategy
505 strategy = mwms_lib.CollectiveAllReduceStrategy(