Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.h32 Status InitializeCollectiveContext(
Dcollective_nccl.cc61 Status NcclBase::InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::NcclBase
Dcollective_nccl_test.cc334 TF_CHECK_OK(reducer->InitializeCollectiveContext(col_ctx)); in RunReduce()
367 TF_CHECK_OK(broadcaster->InitializeCollectiveContext(col_ctx)); in RunBroadcast()
408 TF_CHECK_OK(gatherer->InitializeCollectiveContext(col_ctx)); in RunGather()
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter.h58 Status InitializeCollectiveContext(
Dhierarchical_tree_broadcaster.h42 Status InitializeCollectiveContext(
Dring_alg.h42 Status InitializeCollectiveContext(
Dpermuter.cc53 Status Permuter::InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::Permuter
Dhierarchical_tree_broadcaster.cc187 Status HierarchicalTreeBroadcaster::InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::HierarchicalTreeBroadcaster
Dbase_collective_executor.cc318 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
Dring_alg.cc244 Status RingAlg::InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::RingAlg
Dpermuter_test.cc405 TF_CHECK_OK(permuter->InitializeCollectiveContext(col_ctx)); in DoPermute()
Dring_gatherer_test.cc494 TF_CHECK_OK(gatherer->InitializeCollectiveContext(col_ctx)); in DoGather()
Dring_reducer_test.cc527 TF_CHECK_OK(reducer->InitializeCollectiveContext(col_ctx)); in DoReduce()
Dhierarchical_tree_broadcaster_test.cc688 TF_CHECK_OK(broadcaster->InitializeCollectiveContext(col_ctx)); in DoBroadcast()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h419 virtual Status InitializeCollectiveContext(
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc368 Status InitializeCollectiveContext( in InitializeCollectiveContext() function in tensorflow::__anon53767e020111::__anon53767e020311::MockNcclReducer