Home
last modified time | relevance | path

Searched refs:col_ctx (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext() argument
45 col_ctx_ = col_ctx; in InitializeCollectiveContext()
46 col_params_ = &col_ctx->col_params; in InitializeCollectiveContext()
48 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
49 &col_ctx->device_locality); in InitializeCollectiveContext()
Dcollective_nccl_reducer.h32 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dcollective_nccl_reducer_test.cc291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
294 TF_CHECK_OK(reducer.InitializeCollectiveContext(&col_ctx)); in DoReduce()
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.cc188 CollectiveContext* col_ctx) { in InitializeCollectiveContext() argument
189 CHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
190 col_ctx_ = col_ctx; in InitializeCollectiveContext()
191 col_params_ = &col_ctx->col_params; in InitializeCollectiveContext()
193 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
194 &col_ctx->device_locality); in InitializeCollectiveContext()
Dring_alg.cc243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext() argument
244 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
245 col_ctx_ = col_ctx; in InitializeCollectiveContext()
246 col_params_ = &col_ctx->col_params; in InitializeCollectiveContext()
248 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
249 &col_ctx->device_locality); in InitializeCollectiveContext()
Dbase_collective_executor.cc253 CollectiveContext* col_ctx = in ExecuteAsync() local
256 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
259 delete col_ctx; in ExecuteAsync()
267 SchedClosure([col_impl, col_ctx, done_safe]() { in ExecuteAsync()
268 col_impl->Run([col_impl, col_ctx, done_safe](const Status& s) { in ExecuteAsync()
270 delete col_ctx; in ExecuteAsync()
Dhierarchical_tree_broadcaster.h42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dring_alg.h42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dring_gatherer_test.cc483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather() local
486 TF_CHECK_OK(gatherer.InitializeCollectiveContext(&col_ctx)); in DoGather()
Dring_reducer_test.cc513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce() local
516 TF_CHECK_OK(reducer.InitializeCollectiveContext(&col_ctx)); in DoReduce()
Dhierarchical_tree_broadcaster_test.cc684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast() local
687 TF_CHECK_OK(broadcaster.InitializeCollectiveContext(&col_ctx)); in DoBroadcast()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h384 virtual Status InitializeCollectiveContext(CollectiveContext* col_ctx) = 0;