Home
last modified time | relevance | path

Searched refs:same_num_devices_per_task (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_param_resolver_local_test.cc206 EXPECT_TRUE(cps[i]->group.same_num_devices_per_task); in TEST_F()
260 EXPECT_TRUE(cps[i]->group.same_num_devices_per_task); in TEST_F()
H A Dcollective_param_resolver_local.cc474 gp->same_num_devices_per_task = false; in SetDevPerTask()
483 gp->same_num_devices_per_task = true; in SetDevPerTask()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.h83 bool same_num_devices_per_task = false; member