Searched refs:CollectiveParamResolverDistributed (Results 1 – 6 of 6) sorted by relevance
89 CollectiveParamResolverDistributed::CollectiveParamResolverDistributed( in CollectiveParamResolverDistributed() function in tensorflow::CollectiveParamResolverDistributed107 void CollectiveParamResolverDistributed::CompleteParamsAsync( in CompleteParamsAsync()142 void CollectiveParamResolverDistributed::CompleteGroupAsync( in CompleteGroupAsync()148 void CollectiveParamResolverDistributed::CompleteInstanceAsync( in CompleteInstanceAsync()205 CollectiveParamResolverDistributed::GroupRec*206 CollectiveParamResolverDistributed::GetCachedGroup(int32_t group_key) { in GetCachedGroup()215 Status CollectiveParamResolverDistributed::UpdateGroupCache( in UpdateGroupCache()273 void CollectiveParamResolverDistributed::CompleteGroupDistributed( in CompleteGroupDistributed()316 bool CollectiveParamResolverDistributed::InstanceIsCached( in InstanceIsCached()327 Status CollectiveParamResolverDistributed::UpdateInstanceCache( in UpdateInstanceCache()[all …]
29 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {31 CollectiveParamResolverDistributed(
22 class CollectiveParamResolverDistributed; variable40 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver,
147 std::make_unique<CollectiveParamResolverDistributed>( in DefineWorker()220 CollectiveParamResolverDistributed* cp_res = cp_resolvers_[task_name].get(); in IssueRequest()309 std::unique_ptr<CollectiveParamResolverDistributed>>
53 std::unique_ptr<CollectiveParamResolverDistributed> cpr( in RpcCollectiveExecutorMgrTest()54 new CollectiveParamResolverDistributed( in RpcCollectiveExecutorMgrTest()
31 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in RpcCollectiveExecutorMgr()177 auto param_resolver = std::make_unique<CollectiveParamResolverDistributed>( in CreateProdRpcCollectiveExecutorMgr()