Home
last modified time | relevance | path

Searched defs:num_ranks (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc468 const int num_ranks = this->NumGPUs(); in TYPED_TEST() local
503 const int num_ranks = this->NumGPUs(); in TYPED_TEST() local
578 const int num_ranks = this->NumGPUs(); in TYPED_TEST() local
623 for (int num_ranks = 1; num_ranks <= this->NumGPUs(); ++num_ranks) { in TYPED_TEST() local
687 const int num_ranks = 2; in TYPED_TEST() local
724 const int num_ranks = 2; in TYPED_TEST() local
753 const int num_ranks = 2; in TYPED_TEST() local
781 const int num_ranks = 2; in TYPED_TEST() local
807 const int num_ranks = 2; in TYPED_TEST() local
835 const int num_ranks = 2; in TYPED_TEST() local
/external/coreboot/src/vendorcode/cavium/bdk/libdram/
Ddram-spd.c372 int ddr_interface_num, int num_ranks, int dram_width, int dimm_size_mb) in report_common_dimm()
422 int dimm, int ddr_interface_num, int num_ranks, in report_ddr3_dimm()
462 int dimm, int ddr_interface_num, int num_ranks, in report_ddr4_dimm()
482 int dimm, int ddr_interface_num, int num_ranks, in report_dimm()
Ddram-init-ddr3.c2897 int row_bits, col_bits, num_banks, num_ranks, dram_width; in init_octeon3_ddr3_interface() local
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc111 void Init(const int num_ranks, const int instance_key) { in Init()
187 void RunTest(int num_ranks, int input_length, int instance_key) { in RunTest()
464 const int num_ranks) override { in InitExpected()
499 const int num_ranks) override { in InitExpected()
533 const int num_ranks) override { in InitExpected()
/external/coreboot/src/vendorcode/cavium/bdk/libbdk-dram/
Dbdk-dram-size.c199 int num_ranks = bdk_pop(lmcx_config.s.init_status); in bdk_dram_get_size_mbytes() local
/external/pytorch/torch/cuda/
Dnccl.py60 def init_rank(num_ranks, uid, rank): argument
/external/kernel-headers/original/uapi/linux/
Dkfd_ioctl.h382 __u32 num_ranks; /* from KFD */ member