Searched refs:_get_test_objects (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 115 def _get_test_objects(self, task_type, task_id, num_gpus): member in ParameterServerStrategyTestBase
125 d, _, sess_config = self._get_test_objects(task_type, task_id, num_gpus)
224 d, _, sess_config = self._get_test_objects(task_type, task_id, num_gpus)
373 d, master_target, sess_config = self._get_test_objects(
437 d, master_target, sess_config = self._get_test_objects(
530 distribution, master_target, config = self._get_test_objects(
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 457 def _get_test_objects(self, member in CollectiveAllReduceTest
536 collective_all_reduce, devices, master_target = self._get_test_objects(
675 collective_all_reduce, devices, master_target = self._get_test_objects(
801 collective, devices, _ = self._get_test_objects(
|