Searched refs:CollectiveParamResolverDistributed (Results 1 – 7 of 7) sorted by relevance
91 CollectiveParamResolverDistributed::CollectiveParamResolverDistributed( in CollectiveParamResolverDistributed() function in tensorflow::CollectiveParamResolverDistributed107 void CollectiveParamResolverDistributed::CompleteParamsAsync( in CompleteParamsAsync()131 void CollectiveParamResolverDistributed::CompleteGroupAsync( in CompleteGroupAsync()167 void CollectiveParamResolverDistributed::CompleteInstanceAsync( in CompleteInstanceAsync()222 CollectiveParamResolverDistributed::GroupRec*223 CollectiveParamResolverDistributed::GetCachedGroup(int32 group_key) { in GetCachedGroup()232 Status CollectiveParamResolverDistributed::UpdateGroupCache( in UpdateGroupCache()286 void CollectiveParamResolverDistributed::CompleteGroupDistributed( in CompleteGroupDistributed()330 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 group_key, in InstanceIsCached()341 Status CollectiveParamResolverDistributed::UpdateInstanceCache( in UpdateInstanceCache()[all …]
22 class CollectiveParamResolverDistributed; variable40 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver,
29 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {31 CollectiveParamResolverDistributed(const ConfigProto& config,
53 CollectiveParamResolverDistributed* cpres) in FakeWorker()85 CollectiveParamResolverDistributed* param_resolver_;169 absl::make_unique<CollectiveParamResolverDistributed>( in DefineWorker()228 CollectiveParamResolverDistributed* cp_res = cp_resolvers_[task_name].get(); in IssueRequest()313 std::unique_ptr<CollectiveParamResolverDistributed>>
53 std::unique_ptr<CollectiveParamResolverDistributed> cpr( in RpcCollectiveExecutorMgrTest()54 new CollectiveParamResolverDistributed(options.config, in RpcCollectiveExecutorMgrTest()
31 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in RpcCollectiveExecutorMgr()
284 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver( in Init()285 new CollectiveParamResolverDistributed(config, worker_env_.device_mgr, in Init()453 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver( in UpdateServerDef()454 new CollectiveParamResolverDistributed( in UpdateServerDef()