Searched refs:CollectiveParamResolverLocal (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 41 CollectiveParamResolverLocal::CollectiveParamResolverLocal( in CollectiveParamResolverLocal() function in tensorflow::CollectiveParamResolverLocal
49 void CollectiveParamResolverLocal::CompleteGroupAsync( in CompleteGroupAsync()
87 void CollectiveParamResolverLocal::CompleteGroupLocal( in CompleteGroupLocal()
512 void CollectiveParamResolverLocal::FinishGroup(GroupRec* gr) { in FinishGroup()
535 void CollectiveParamResolverLocal::CompleteTaskIsLocal(const string& task_name, in CompleteTaskIsLocal()
543 void CollectiveParamResolverLocal::SetDefaultRank(const string& device, in SetDefaultRank()
554 void CollectiveParamResolverLocal::InitInstanceSharedParams( in InitInstanceSharedParams()
570 void CollectiveParamResolverLocal::CompleteDefaultRanking( in CompleteDefaultRanking()
599 CollectiveParamResolverLocal::InstanceRec*
600 CollectiveParamResolverLocal::GetOrCreateInstanceRec(const GroupRec* gr, in GetOrCreateInstanceRec()
[all …]
|
| D | collective_param_resolver_local.h | 41 class CollectiveParamResolverLocal : public ParamResolverInterface {
43 CollectiveParamResolverLocal(const ConfigProto& config,
48 ~CollectiveParamResolverLocal() override {} in ~CollectiveParamResolverLocal()
|
| D | collective_rma_local_test.cc | 51 prl_ = absl::make_unique<CollectiveParamResolverLocal>( in CollectiveRemoteAccessLocalTest()
63 std::unique_ptr<CollectiveParamResolverLocal> prl_;
|
| D | collective_executor_mgr_test.cc | 48 new CollectiveParamResolverLocal(cp, device_mgr_.get(), drl.get(), in CollectiveExecutorMgrTest()
|
| D | collective_param_resolver_local_test.cc | 57 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in ResetParamResolver()
88 std::unique_ptr<CollectiveParamResolverLocal> prl_;
|
| D | direct_session.cc | 554 new CollectiveParamResolverLocal(options_.config, device_mgr_.get(), in RunInternal()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.h | 29 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {
|
| D | collective_param_resolver_distributed.cc | 95 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 121 std::unique_ptr<ParamResolverInterface> cprl(new CollectiveParamResolverLocal(
|