HomeSort by relevance Sort by last modified time
    Searched refs:CollectiveContext (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.h 32 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
42 CollectiveContext* col_ctx_; // Not owned
collective_nccl_reducer.cc 44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) {
collective_nccl_reducer_test.cc 291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(),
  /external/tensorflow/tensorflow/core/common_runtime/
hierarchical_tree_broadcaster.h 40 // Initializes members of CollectiveContext not yet initialized, i.e. device
41 // and device_locality. Also saves the CollectiveContext in this object.
42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
82 CollectiveContext* col_ctx_; // Not owned
ring_alg.h 40 // Initializes members of CollectiveContext not yet initialized, i.e. device
41 // and device_locality. Also saves the CollectiveContext in this object.
42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
110 CollectiveContext* col_ctx_; // Not owned
base_collective_executor.cc 253 CollectiveContext* col_ctx =
254 new CollectiveContext(this, dev_mgr_, ctx, CtxParams(ctx), col_params,
hierarchical_tree_broadcaster.cc 188 CollectiveContext* col_ctx) {
ring_alg.cc 243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) {
hierarchical_tree_broadcaster_test.cc 684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(),
    [all...]
ring_gatherer_test.cc 483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(),
ring_reducer_test.cc 513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(),
  /external/tensorflow/tensorflow/core/framework/
collective.h 339 class CollectiveContext {
341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
346 virtual ~CollectiveContext() = default;
380 // Prepares the CollectiveContext for executing this CollectiveImplementation.
382 // CollectiveContext passed in must outlive the CollectiveImplementation
384 virtual Status InitializeCollectiveContext(CollectiveContext* col_ctx) = 0;
collective.cc 156 CollectiveContext::CollectiveContext(CollectiveExecutor* col_exec,

Completed in 464 milliseconds