Searched refs:TpuCompilationCacheRpcLookup (Results 1 – 3 of 3) sorted by relevance
50 TpuCompilationCacheRpcLookup::TpuCompilationCacheRpcLookup( in TpuCompilationCacheRpcLookup() function in tensorflow::tpu::TpuCompilationCacheRpcLookup63 Status TpuCompilationCacheRpcLookup::Lookup( in Lookup()98 Status TpuCompilationCacheRpcLookup::Lookup( in Lookup()143 Status TpuCompilationCacheRpcLookup::RemoteLookupLocked( in RemoteLookupLocked()170 void TpuCompilationCacheRpcLookup::PostLookupLocked( in PostLookupLocked()197 std::string TpuCompilationCacheRpcLookup::DebugString() const { in DebugString()
36 class TpuCompilationCacheRpcLookup : public TpuCompilationCacheLookup {40 TpuCompilationCacheRpcLookup(const string& server_address,42 ~TpuCompilationCacheRpcLookup() override = default;
351 new tpu::TpuCompilationCacheRpcLookup(server_address, cache_size_bytes); in Compute()