Home
last modified time | relevance | path

Searched refs:channel_cache (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache_test.cc34 auto channel_cache = std::shared_ptr<GrpcChannelCache>( in TEST() local
41 NewGrpcWorkerCache(channel_cache, grpc_worker_env.get())); in TEST()
59 channel_cache, grpc_worker_env.get(), local_wi.get(), "local_target")); in TEST()
69 auto channel_cache = std::shared_ptr<GrpcChannelCache>( in TEST() local
77 NewGrpcWorkerCache(channel_cache, grpc_worker_env.get()); in TEST()
Dgrpc_worker_cache.cc35 explicit GrpcWorkerCache(std::shared_ptr<GrpcChannelCache> channel_cache, in GrpcWorkerCache() argument
41 channel_cache_(channel_cache), in GrpcWorkerCache()
Dgrpc_server_lib.cc385 std::shared_ptr<GrpcChannelCache> channel_cache(NewGrpcChannelCache( in WorkerCacheFactory() local
391 const string host_port = channel_cache->TranslateTask(name_prefix); in WorkerCacheFactory()
405 channel_cache, grpc_worker_env(), worker_impl(), name_prefix); in WorkerCacheFactory()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client_test.cc34 auto channel_cache = std::shared_ptr<GrpcChannelCache>( in TEST() local
37 NewGrpcEagerClientCache(channel_cache)); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/coordination/
Dgrpc_coordination_client.cc229 std::shared_ptr<GrpcChannelCache> channel_cache) in GrpcCoordinationClientCache() argument
231 channel_cache_(channel_cache), in GrpcCoordinationClientCache()
291 std::shared_ptr<GrpcChannelCache> channel_cache) { in NewGrpcCoordinationClientCache() argument
292 return new GrpcCoordinationClientCache(channel_cache); in NewGrpcCoordinationClientCache()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device_test.cc53 std::shared_ptr<GrpcChannelCache> channel_cache( in RemoteDeviceTest() local
57 NewGrpcWorkerCache(channel_cache, grpc_worker_env_.get())); in RemoteDeviceTest()
Dcluster_function_library_runtime_test.cc40 std::shared_ptr<GrpcChannelCache> channel_cache( in ClusterFunctionLibraryRuntimeTest() local
43 NewGrpcWorkerCache(channel_cache, grpc_worker_env_.get())); in ClusterFunctionLibraryRuntimeTest()