Home
last modified time | relevance | path

Searched refs:CollectiveParamResolverDistributed (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_param_resolver_distributed.cc89 CollectiveParamResolverDistributed::CollectiveParamResolverDistributed( in CollectiveParamResolverDistributed() function in tensorflow::CollectiveParamResolverDistributed
107 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 …]
H A Dcollective_param_resolver_distributed.h29 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {
31 CollectiveParamResolverDistributed(
H A Drpc_collective_executor_mgr.h22 class CollectiveParamResolverDistributed; variable
40 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver,
H A Dcollective_param_resolver_distributed_test.cc147 std::make_unique<CollectiveParamResolverDistributed>( in DefineWorker()
220 CollectiveParamResolverDistributed* cp_res = cp_resolvers_[task_name].get(); in IssueRequest()
309 std::unique_ptr<CollectiveParamResolverDistributed>>
H A Drpc_collective_executor_mgr_test.cc53 std::unique_ptr<CollectiveParamResolverDistributed> cpr( in RpcCollectiveExecutorMgrTest()
54 new CollectiveParamResolverDistributed( in RpcCollectiveExecutorMgrTest()
H A Drpc_collective_executor_mgr.cc31 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in RpcCollectiveExecutorMgr()
177 auto param_resolver = std::make_unique<CollectiveParamResolverDistributed>( in CreateProdRpcCollectiveExecutorMgr()