| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| imagenet_model_evaluator.h | 32 // controlled by |num_ranks|.
72 int num_ranks = 10; member in struct:tensorflow::metrics::ImagenetModelEvaluator::Params
|
| /external/tensorflow/tensorflow/core/nccl/ |
| nccl_manager_test.cc | 260 const int num_ranks = 4; local
265 this->MakeReductionTestCase(/*num_nodes=*/1, num_ranks, reduction_op,
267 for (int rank = 0; rank < num_ranks; ++rank) {
278 {"allreduce", /*num_local_devices=*/num_ranks,
279 /*num_global_devices=*/num_ranks, /*communicator_key=*/""},
291 // To run test longer, increase num_ranks, num_collectives_per_iteration and
294 const int num_ranks = 4; local
296 const int num_threads = num_ranks * 2;
307 /*num_nodes=*/1, num_ranks, ncclSum,
309 for (int j = 0; j < num_ranks; ++j)
373 const int num_ranks = 4; local
466 const int num_ranks = 2; local
501 const int num_ranks = 2; local
529 const int num_ranks = 2; local
[all...] |
| /bionic/libc/kernel/uapi/linux/ |
| kfd_ioctl.h | 219 __u32 num_ranks; member in struct:kfd_ioctl_get_tile_config_args
|
| /external/kernel-headers/original/uapi/linux/ |
| kfd_ioctl.h | 308 __u32 num_ranks; /* from KFD */ member in struct:kfd_ioctl_get_tile_config_args
|
