Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.h32 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dcollective_nccl_reducer.cc44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext() function in tensorflow::NcclReducer
Dcollective_nccl_reducer_test.cc294 TF_CHECK_OK(reducer.InitializeCollectiveContext(&col_ctx)); in DoReduce()
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.h42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dring_alg.h42 Status InitializeCollectiveContext(CollectiveContext* col_ctx) override;
Dbase_collective_executor.cc256 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
Dhierarchical_tree_broadcaster.cc187 Status HierarchicalTreeBroadcaster::InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::HierarchicalTreeBroadcaster
Dring_alg.cc243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext() function in tensorflow::RingAlg
Dring_gatherer_test.cc486 TF_CHECK_OK(gatherer.InitializeCollectiveContext(&col_ctx)); in DoGather()
Dring_reducer_test.cc516 TF_CHECK_OK(reducer.InitializeCollectiveContext(&col_ctx)); in DoReduce()
Dhierarchical_tree_broadcaster_test.cc687 TF_CHECK_OK(broadcaster.InitializeCollectiveContext(&col_ctx)); in DoBroadcast()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h384 virtual Status InitializeCollectiveContext(CollectiveContext* col_ctx) = 0;