Searched refs:CollectiveParamResolverDistributed (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 81 CollectiveParamResolverDistributed::CollectiveParamResolverDistributed( in CollectiveParamResolverDistributed() function in tensorflow::CollectiveParamResolverDistributed
95 void CollectiveParamResolverDistributed::CompleteParamsAsync( in CompleteParamsAsync()
112 void CollectiveParamResolverDistributed::CompleteGroupAsync( in CompleteGroupAsync()
144 void CollectiveParamResolverDistributed::CompleteInstanceAsync( in CompleteInstanceAsync()
209 bool CollectiveParamResolverDistributed::GroupIsCached(int32 group_key) { in GroupIsCached()
215 Status CollectiveParamResolverDistributed::UpdateGroupCache( in UpdateGroupCache()
253 void CollectiveParamResolverDistributed::CompleteGroupDistributed( in CompleteGroupDistributed()
284 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 instance_key) { in InstanceIsCached()
290 void CollectiveParamResolverDistributed::UpdateInstanceCache( in UpdateInstanceCache()
362 void CollectiveParamResolverDistributed::CompleteInstanceDistributed( in CompleteInstanceDistributed()
|
| D | collective_param_resolver_distributed_test.cc | 50 CollectiveParamResolverDistributed* cpres) in FakeWorker()
82 CollectiveParamResolverDistributed* param_resolver_;
174 CollectiveParamResolverDistributed* cp_res = in DefineWorker()
175 new CollectiveParamResolverDistributed(config, dev_mgr, dev_res, &wc_, in DefineWorker()
233 CollectiveParamResolverDistributed* cp_res = cp_resolvers_[task_name]; in IssueRequest()
287 std::unordered_map<string, CollectiveParamResolverDistributed*> cp_resolvers_;
|
| D | rpc_collective_executor_mgr.h | 22 class CollectiveParamResolverDistributed; variable
40 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver,
|
| D | collective_param_resolver_distributed.h | 26 class CollectiveParamResolverDistributed : public CollectiveParamResolverLocal {
28 CollectiveParamResolverDistributed(const ConfigProto& config,
|
| D | rpc_collective_executor_mgr_test.cc | 50 std::unique_ptr<CollectiveParamResolverDistributed> cpr( in RpcCollectiveExecutorMgrTest()
51 new CollectiveParamResolverDistributed(options.config, in RpcCollectiveExecutorMgrTest()
|
| D | rpc_collective_executor_mgr.cc | 31 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in RpcCollectiveExecutorMgr()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_server_lib.cc | 74 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver( in Init()
75 new CollectiveParamResolverDistributed( in Init()
|
| D | gdr_collective_executor_mgr.h | 38 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in GdrCollectiveExecutorMgr()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 232 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver( in Init()
233 new CollectiveParamResolverDistributed(config, worker_env_.device_mgr, in Init()
|