Searched refs:ds_test_util (Results 1 – 1 of 1) sorted by relevance
37 from tensorflow.python.distribute import test_util as ds_test_util unknown134 ds_test_util.gather(distribution, distributed_values),154 ds_test_util.gather(distribution, distributed_values).numpy(),172 distributed_values = ds_test_util.gather(distribution, distributed_values)197 distributed_values = ds_test_util.gather(distribution, distributed_values)242 distributed_values = ds_test_util.gather(distribution, distributed_values)267 outputs = ds_test_util.gather(1504 ds_test_util.main()