Searched refs:GrpcTpuDriver (Results 1 – 1 of 1) sorted by relevance
46 class GrpcTpuDriver;172 explicit GrpcTpuStream(int32_t id, GrpcTpuDriver* driver,215 friend class GrpcTpuDriver;276 GrpcTpuDriver* driver_;314 class GrpcTpuDriver : public TpuDriver { class316 explicit GrpcTpuDriver(const TpuDriverConfig& config, in GrpcTpuDriver() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver334 ~GrpcTpuDriver() override { in ~GrpcTpuDriver()458 GrpcTpuStream::GrpcTpuStream(int32_t id, GrpcTpuDriver* driver, in GrpcTpuStream()947 GrpcTpuDriver::CreateTpuDriverStub( in CreateTpuDriverStub()989 std::unique_ptr<GrpcTpuStream> GrpcTpuDriver::AllocateStream(int32_t id) { in AllocateStream()[all …]