Searched refs:cache_service (Results 1 – 2 of 2) sorted by relevance
64 auto cache_service = absl::make_unique<TpuCompilationCacheService>( in ConstructCacheService() local66 cache_service->SetMemoryQuota(1ul << 31); // 2GB in ConstructCacheService()67 cache_service->Start(); in ConstructCacheService()68 return cache_service; in ConstructCacheService()99 int service_port, std::unique_ptr<TpuCompilationCacheService> cache_service) in TpuPodState() argument100 : cache_service_(std::move(cache_service)), service_port_(service_port) {} in TpuPodState()179 std::unique_ptr<TpuCompilationCacheService> cache_service, in ConstructTpuPodState()186 new TpuPodState(serving_port, std::move(cache_service))); in ConstructTpuPodState()
36 std::unique_ptr<TpuCompilationCacheService> cache_service);