Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer.h36 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_reducer.h42 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dpermuter.h52 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams() function
Dhierarchical_tree_broadcaster.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_gatherer.cc45 Status RingGatherer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingGatherer
60 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
Dring_alg.h38 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dring_reducer.cc48 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingReducer
52 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
Dcollective_param_resolver_local.cc775 s = col_impl->InitializeCollectiveParams(cp); in CompleteInstanceFromInitializedIRec()
780 done(col_impl->InitializeCollectiveParams(cp)); in CompleteInstanceFromInitializedIRec()
Dhierarchical_tree_broadcaster.cc77 Status HierarchicalTreeBroadcaster::InitializeCollectiveParams( in InitializeCollectiveParams() function in tensorflow::HierarchicalTreeBroadcaster
Dring_alg.cc165 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::RingAlg
Dring_gatherer_test.cc380 TF_CHECK_OK(gatherer->InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
Dring_reducer_test.cc405 TF_CHECK_OK(reducer->InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
Dhierarchical_tree_broadcaster_test.cc532 TF_CHECK_OK(broadcaster->InitializeCollectiveParams(cp)); in RunSubdivPermsTest()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.h29 Status InitializeCollectiveParams(CollectiveParams* col_params) override;
Dcollective_nccl.cc29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams() function in tensorflow::NcclBase
/external/tensorflow/tensorflow/core/framework/
Dcollective.h413 virtual Status InitializeCollectiveParams(CollectiveParams* col_params) = 0;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc365 Status InitializeCollectiveParams(CollectiveParams*) override { in InitializeCollectiveParams() function in tensorflow::__anon53767e020111::__anon53767e020311::MockNcclReducer