Searched refs:SetMemoryQuota (Results 1 – 3 of 3) sorted by relevance
45 void SetMemoryQuota(size_t max_bytes);
81 void TpuCompilationCacheService::SetMemoryQuota(size_t max_bytes) { in SetMemoryQuota() function in tensorflow::TpuCompilationCacheService
66 cache_service->SetMemoryQuota(1ul << 31); // 2GB in ConstructCacheService()