Searched refs:selectAllReduceAlgo (Results 1 – 3 of 3) sorted by relevance
66 AllReduceAlgo selectAllReduceAlgo(const at::Tensor& input);
675 AllReduceAlgo IntraNodeComm::selectAllReduceAlgo(const at::Tensor& input) { in selectAllReduceAlgo() function in c10d::intra_node_comm::IntraNodeComm
3453 auto algo = intraNodeComm_->selectAllReduceAlgo(tensor); in checkForNCCLErrorsInternal()