Searched defs:num_subdivs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 114 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_reducer_test.cc | 156 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
285 int num_devices, int num_subdivs, int tensor_len, in RunTest()
419 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance() local
630 int num_subdivs = 2; in TEST_F() local
|
| D | ring_gatherer_test.cc | 134 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
263 int num_devices, int num_subdivs, int tensor_len, in RunTest()
395 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance() local
|
| D | ring_alg.cc | 124 int num_subdivs = 0; in GenerateSubdivsInCollectiveParams() local
|
| D | hierarchical_tree_broadcaster_test.cc | 266 int num_subdivs = num_workers + (num_workers > 1 ? 1 : 0); in Init() local
583 size_t num_subdivs = impl.subdiv_permutations.size(); in DeviceInstance() local
|