Searched defs:num_subdivs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | ring_reducer_test.cc | 86 int num_subdivs, int fail_after) { in Init()
117 int num_devices, int num_subdivs, int tensor_len, in RunTest()
159 DeviceInstance(int rank, int num_subdivs, DataType dtype, in DeviceInstance()
283 int num_subdivs = 2; in TEST_F() local
|
| H A D | ring_gatherer_test.cc | 49 int num_subdivs, int fail_after) { in Init()
80 int num_devices, int num_subdivs, int tensor_len, in RunTest()
124 DeviceInstance(int rank, int num_subdivs, DataType dtype, in DeviceInstance()
|
| H A D | hierarchical_tree_broadcaster.cc | 114 int num_subdivs = num_tasks + (num_tasks > 1 ? 1 : 0); in InitializeCollectiveParams() local
292 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in RunTree() local
|
| H A D | ring_alg.cc | 140 int num_subdivs = 0; in GenerateSubdivsInCollectiveParams() local
|
| H A D | collective_test_util.cc | 252 int num_subdivs) { in GenerateEvenSubdivOffsets()
|