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)
793 expected_input_pipeline_id=None)
Dmirrored_strategy_test.py170 expected_input_pipeline_id=0)
181 expected_input_pipeline_id=0)
198 expected_input_pipeline_id=0)
1204 expected_input_pipeline_id=None)
1229 expected_input_pipeline_id=None)
Dcollective_all_reduce_strategy_test.py365 expected_input_pipeline_id=task_id)
464 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.py999 expected_input_pipeline_id=0)
1010 expected_input_pipeline_id=0)