Searched refs:channel_cache (Results 1 – 9 of 9) sorted by relevance
52 GrpcChannelCache** channel_cache) { in ChannelCacheFactory() argument60 *channel_cache = in ChannelCacheFactory()63 const string host_port = (*channel_cache)->TranslateTask(name_prefix); in ChannelCacheFactory()69 (*channel_cache)->TranslateTask(name_prefix), in ChannelCacheFactory()
36 GrpcChannelCache* const channel_cache);
47 GrpcChannelCache** channel_cache);
36 GrpcChannelCache* const channel_cache) in RdmaMgr() argument37 : worker_env_(worker_env), channel_cache_(channel_cache) { in RdmaMgr()
52 std::shared_ptr<GrpcChannelCache> channel_cache( in RemoteDeviceTest() local54 worker_cache_.reset(NewGrpcWorkerCache(channel_cache)); in RemoteDeviceTest()
39 explicit GrpcWorkerCache(std::shared_ptr<GrpcChannelCache> channel_cache, in GrpcWorkerCache() argument44 channel_cache_(channel_cache), in GrpcWorkerCache()
98 std::shared_ptr<GrpcChannelCache> channel_cache() { return channel_cache_; } in channel_cache() function
48 GrpcChannelCache** channel_cache);
201 std::shared_ptr<tensorflow::GrpcChannelCache> channel_cache = in UpdateTFE_ContextWithServerDef() local202 grpc_server->channel_cache(); in UpdateTFE_ContextWithServerDef()204 tensorflow::eager::NewGrpcEagerClientCache(channel_cache)); in UpdateTFE_ContextWithServerDef()