Home
last modified time | relevance | path

Searched defs:dev_per_task (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dhierarchical_tree_broadcaster.cc64 int device_rank, const std::vector<int>& dev_per_task) { in GetDeviceTask()
88 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
H A Dring_alg.cc192 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
H A Dhierarchical_tree_broadcaster_test.cc358 std::vector<int> dev_per_task = {4, 4, 6, 8}; in TEST_F() local
H A Dcollective_param_resolver_local.cc475 int dev_per_task = -1; in SetDevPerTask() local