Searched refs:GetTpuProgram (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_grpc.h | 73 virtual ::grpc::Status GetTpuProgram(::grpc::ClientContext* context,
106 ::grpc::Status GetTpuProgram(::grpc::ClientContext* context,
145 virtual ::grpc::Status GetTpuProgram(::grpc::ServerContext* context,
160 ::grpc::Status GetTpuProgram(::grpc::ServerContext* context, in GetTpuProgram() function
190 ::grpc::Status GetTpuProgram(::grpc::ServerContext* context, in GetTpuProgram() function
215 ::grpc::Status GetTpuProgram(::grpc::ServerContext* context, in GetTpuProgram() function
|
| D | tpu_compilation_cache_grpc.cc | 58 ::grpc::Status grpc::TpuCompilationCacheService::Stub::GetTpuProgram( in GetTpuProgram() function in tensorflow::tpu::grpc::TpuCompilationCacheService::Stub
92 &grpc::TpuCompilationCacheService::Service::GetTpuProgram), in Service()
98 ::grpc::Status grpc::TpuCompilationCacheService::Service::GetTpuProgram( in GetTpuProgram() function in tensorflow::tpu::grpc::TpuCompilationCacheService::Service
|
| D | tpu_compilation_cache.proto | 22 // Response for GetTpuProgram RPC.
40 rpc GetTpuProgram(GetTpuProgramRequest) rpc
|
| D | tpu_compilation_cache_service.cc | 89 void TpuCompilationCacheService::GetTpuProgram(GetTpuProgramCall* call) { in GetTpuProgram() function in tensorflow::TpuCompilationCacheService
144 thread_pool_->Schedule([this, call]() { GetTpuProgram(call); }); in HandleGetTpuProgram()
|
| D | tpu_compilation_cache_service.h | 57 void GetTpuProgram(GetTpuProgramCall* call);
|
| D | tpu_compilation_cache_rpc_lookup.cc | 157 FromGrpcStatus(stub_->GetTpuProgram(&client_context, request, &response)); in RemoteLookupLocked()
|