Searched refs:create_test_objects (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | parameter_server_strategy_test.py | 69 def create_test_objects(cluster_spec=None, function
112 return create_test_objects(
578 strategy, _, _ = create_test_objects(num_gpus=2)
585 strategy, _, _ = create_test_objects(num_gpus=0)
591 strategy, _, _ = create_test_objects(num_gpus=1)
597 strategy, _, _ = create_test_objects(num_gpus=2)
697 strategy, _, _ = create_test_objects()
702 strategy, _, _ = create_test_objects(
721 strategy, _, _ = create_test_objects(num_gpus=2)
731 strategy, _, _ = create_test_objects(
[all …]
|
| H A D | collective_all_reduce_strategy_test.py | 71 def create_test_objects(cluster_spec=None, function
116 strategy, target = create_test_objects(
298 distribution, _ = create_test_objects(
|