Searched refs:TpuCompilationCacheService (Results 1 – 8 of 8) sorted by relevance
41 std::unique_ptr<grpc::TpuCompilationCacheService::Stub>42 grpc::TpuCompilationCacheService::NewStub( in NewStub()46 std::unique_ptr<grpc::TpuCompilationCacheService::Stub> stub( in NewStub()47 new grpc::TpuCompilationCacheService::Stub(channel)); in NewStub()51 grpc::TpuCompilationCacheService::Stub::Stub( in Stub()58 ::grpc::Status grpc::TpuCompilationCacheService::Stub::GetTpuProgram( in GetTpuProgram()66 grpc::TpuCompilationCacheService::ResponseType>*67 grpc::TpuCompilationCacheService::Stub::AsyncGetTpuProgramRaw( in AsyncGetTpuProgramRaw()76 grpc::TpuCompilationCacheService::ResponseType>*77 grpc::TpuCompilationCacheService::Stub::PrepareAsyncGetTpuProgramRaw( in PrepareAsyncGetTpuProgramRaw()[all …]
33 TpuCompilationCacheService::TpuCompilationCacheService( in TpuCompilationCacheService() function in tensorflow::TpuCompilationCacheService46 TpuCompilationCacheService::~TpuCompilationCacheService() { in ~TpuCompilationCacheService()59 void TpuCompilationCacheService::Start() { in Start()66 bool TpuCompilationCacheService::Shutdown(int timeout_sec) { in Shutdown()81 void TpuCompilationCacheService::SetMemoryQuota(size_t max_bytes) { in SetMemoryQuota()89 void TpuCompilationCacheService::GetTpuProgram(GetTpuProgramCall* call) { in GetTpuProgram()143 void TpuCompilationCacheService::HandleGetTpuProgram(GetTpuProgramCall* call) { in HandleGetTpuProgram()149 &TpuCompilationCacheService::HandleGetTpuProgram, in HandleGetTpuProgram()154 void TpuCompilationCacheService::HandleRPCsLoop() { in HandleRPCsLoop()162 &TpuCompilationCacheService::HandleGetTpuProgram, in HandleRPCsLoop()[all …]
34 class TpuCompilationCacheService {36 using ServiceType = ::tensorflow::tpu::grpc::TpuCompilationCacheService;39 TpuCompilationCacheService(::grpc::ServerBuilder* server_builder,41 ~TpuCompilationCacheService();50 using GetTpuProgramCall = Call<TpuCompilationCacheService, AsyncService,
53 xla::StatusOr<std::unique_ptr<TpuCompilationCacheService>>64 auto cache_service = absl::make_unique<TpuCompilationCacheService>( in ConstructCacheService()99 int service_port, std::unique_ptr<TpuCompilationCacheService> cache_service) in TpuPodState()179 std::unique_ptr<TpuCompilationCacheService> cache_service, in ConstructTpuPodState()
36 std::unique_ptr<TpuCompilationCacheService> cache_service);43 std::unique_ptr<TpuCompilationCacheService> cache_service_;
38 using StubType = tpu::grpc::TpuCompilationCacheService::Stub;
59 stub_ = tpu::grpc::TpuCompilationCacheService::NewStub(channel); in TpuCompilationCacheRpcLookup()
48 class TpuCompilationCacheService final {