Searched refs:CollectiveContext (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.h | 32 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
42 CollectiveContext* col_ctx_; // Not owned
|
| D | collective_nccl_reducer.cc | 44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
|
| D | collective_nccl_reducer_test.cc | 291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.h | 42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
82 CollectiveContext* col_ctx_; // Not owned
|
| D | ring_alg.h | 42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
110 CollectiveContext* col_ctx_; // Not owned
|
| D | base_collective_executor.cc | 253 CollectiveContext* col_ctx = in ExecuteAsync()
254 new CollectiveContext(this, dev_mgr_, ctx, CtxParams(ctx), col_params, in ExecuteAsync()
|
| D | hierarchical_tree_broadcaster.cc | 188 CollectiveContext* col_ctx) { in InitializeCollectiveContext()
|
| D | ring_alg.cc | 243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
|
| D | ring_gatherer_test.cc | 483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather()
|
| D | ring_reducer_test.cc | 513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 339 class CollectiveContext {
341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
346 virtual ~CollectiveContext() = default;
384 virtual Status InitializeCollectiveContext(CollectiveContext* col_ctx) = 0;
|
| D | collective.cc | 156 CollectiveContext::CollectiveContext(CollectiveExecutor* col_exec, in CollectiveContext() function in tensorflow::CollectiveContext
|