Searched refs:create_test_objects (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 73 def create_test_objects(cluster_spec=None, function
116 return create_test_objects(
582 strategy, _, _ = create_test_objects(num_gpus=2)
589 strategy, _, _ = create_test_objects(num_gpus=0)
595 strategy, _, _ = create_test_objects(num_gpus=1)
601 strategy, _, _ = create_test_objects(num_gpus=2)
701 strategy, _, _ = create_test_objects()
706 strategy, _, _ = create_test_objects(
725 strategy, _, _ = create_test_objects(num_gpus=2)
735 strategy, _, _ = create_test_objects(
[all …]
|
| D | collective_all_reduce_strategy_test.py | 69 def create_test_objects(cluster_spec=None, function
106 strategy, target, session_config = create_test_objects(
273 distribution, _, _ = create_test_objects(
|