Searched defs:num_subdivs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer_test.cc | 163 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
298 int num_devices, int num_subdivs, int tensor_len, in RunTest()
430 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in DeviceInstance() local
647 int num_subdivs = 2; in TEST_F() local
|
| D | hierarchical_tree_broadcaster.cc | 115 int num_subdivs = num_tasks + (num_tasks > 1 ? 1 : 0); in InitializeCollectiveParams() local
291 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in RunTree() local
|
| D | ring_gatherer_test.cc | 140 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
274 int num_devices, int num_subdivs, int tensor_len, in RunTest()
404 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in DeviceInstance() local
|
| D | ring_alg.cc | 125 int num_subdivs = 0; in GenerateSubdivsInCollectiveParams() local
|
| D | hierarchical_tree_broadcaster_test.cc | 280 int num_subdivs = num_workers + (num_workers > 1 ? 1 : 0); in Init() local
585 size_t num_subdivs = impl.subdiv_permutations.size(); in DeviceInstance() local
|