Home
last modified time | relevance | path

Searched refs:InitializeCollectiveParams (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer.h36 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_gatherer.cc43 Status RingGatherer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingGatherer
58 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
Dring_reducer.h42 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dhierarchical_tree_broadcaster.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_reducer.cc46 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingReducer
50 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
Dring_alg.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dcollective_param_resolver_local.cc705 s = col_impl->InitializeCollectiveParams(cp); in CompleteInstanceFromInitializedIRec()
710 done(col_impl->InitializeCollectiveParams(cp)); in CompleteInstanceFromInitializedIRec()
Dhierarchical_tree_broadcaster.cc76 Status HierarchicalTreeBroadcaster::InitializeCollectiveParams( in InitializeCollectiveParams() function in tensorflow::HierarchicalTreeBroadcaster
Dring_alg.cc164 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingAlg
Dring_gatherer_test.cc370 TF_CHECK_OK(gatherer.InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
Dring_reducer_test.cc393 TF_CHECK_OK(reducer.InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
Dhierarchical_tree_broadcaster_test.cc526 TF_CHECK_OK(broadcaster.InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.h29 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dcollective_nccl_reducer.cc31 Status NcclReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::NcclReducer
/external/tensorflow/tensorflow/core/framework/
Dcollective.h378 virtual Status InitializeCollectiveParams(CollectiveParams* col_params) = 0;