Searched defs:col_ctx (Results 1 – 8 of 8) sorted by relevance
44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
253 CollectiveContext* col_ctx = in ExecuteAsync() local
188 CollectiveContext* col_ctx) { in InitializeCollectiveContext()
483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather() local
513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast() local