Searched refs:CollectiveParamResolverLocal (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 36 void CollectiveParamResolverLocal::InstanceRec::WaitForOutMu(mutex_lock& lock) { in WaitForOutMu()
40 CollectiveParamResolverLocal::CollectiveParamResolverLocal( in CollectiveParamResolverLocal() function in tensorflow::CollectiveParamResolverLocal
48 void CollectiveParamResolverLocal::CompleteGroupAsync( in CompleteGroupAsync()
57 void CollectiveParamResolverLocal::CompleteGroupLocal( in CompleteGroupLocal()
381 void CollectiveParamResolverLocal::CompleteTaskIsLocal(const string& task_name, in CompleteTaskIsLocal()
389 void CollectiveParamResolverLocal::SetDefaultRank(const string& device, in SetDefaultRank()
400 void CollectiveParamResolverLocal::InitInstanceSharedParams( in InitInstanceSharedParams()
459 void CollectiveParamResolverLocal::CompleteDefaultRanking( in CompleteDefaultRanking()
490 void CollectiveParamResolverLocal::CallbackWithStatus( in CallbackWithStatus()
501 void CollectiveParamResolverLocal::FindInstanceRec( in FindInstanceRec()
[all …]
|
| D | collective_param_resolver_local.h | 38 class CollectiveParamResolverLocal : public ParamResolverInterface {
40 CollectiveParamResolverLocal(const ConfigProto& config,
45 ~CollectiveParamResolverLocal() override {} in ~CollectiveParamResolverLocal()
|
| D | collective_rma_local_test.cc | 49 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in CollectiveRemoteAccessLocalTest()
57 std::unique_ptr<CollectiveParamResolverLocal> prl_;
|
| D | collective_param_resolver_local_test.cc | 44 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in CollectiveParamResolverLocalTest()
55 CollectiveParamResolverLocal::InstanceRec ir; in RunCompleteDefaultRanking()
79 std::unique_ptr<CollectiveParamResolverLocal> prl_;
|
| D | collective_executor_mgr_test.cc | 47 new CollectiveParamResolverLocal(cp, device_mgr_.get(), drl.get(), in CollectiveExecutorMgrTest()
|
| D | direct_session.cc | 504 new CollectiveParamResolverLocal(options_.config, device_mgr_.get(), in RunInternal()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.h | 26 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {
|
| D | collective_param_resolver_distributed.cc | 85 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 86 std::unique_ptr<ParamResolverInterface> cprl(new CollectiveParamResolverLocal( in EagerContext()
|