Home
last modified time | relevance | path

Searched refs:expected_input_pipeline_id (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py64 expected_input_pipeline_id=0)
74 expected_input_pipeline_id=0)
89 expected_input_pipeline_id=0)
Dstrategy_test_lib.py315 expected_input_pipeline_id): argument
329 if expected_input_pipeline_id is not None:
330 self.assertEqual(expected_input_pipeline_id,
768 expected_input_pipeline_id=None)
794 expected_input_pipeline_id=None)
Dmirrored_strategy_test.py197 expected_input_pipeline_id=0)
208 expected_input_pipeline_id=0)
225 expected_input_pipeline_id=0)
1237 expected_input_pipeline_id=None)
1262 expected_input_pipeline_id=None)
Dcollective_all_reduce_strategy_test.py366 expected_input_pipeline_id=task_id)
465 expected_input_pipeline_id=0)
Dparameter_server_strategy_test.py658 expected_input_pipeline_id=1) # because task_id = 1
689 expected_input_pipeline_id=0) # only one worker and pipeline for local.
Dtpu_strategy_test.py1148 expected_input_pipeline_id=0)
1159 expected_input_pipeline_id=0)