nccl_manager.h | 60 Tensor* output, int global_rank, DoneCallback done_callback) 67 global_rank(global_rank), 102 // `global_rank` is not required for single-node collectives. 103 const int global_rank; member in struct:tensorflow::NcclManager::Participant
|