Home
last modified time | relevance | path

Searched defs:col_ctx (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
Dcollective_nccl_reducer_test.cc291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
Dbase_collective_executor.cc253 CollectiveContext* col_ctx = in ExecuteAsync() local
Dhierarchical_tree_broadcaster.cc188 CollectiveContext* col_ctx) { in InitializeCollectiveContext()
Dring_gatherer_test.cc483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather() local
Dring_reducer_test.cc513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
Dhierarchical_tree_broadcaster_test.cc684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast() local