Searched refs:CreateTpuDriverStub (Results 1 – 1 of 1) sorted by relevance
422 static std::unique_ptr<grpc::CloudTpuDriver::Stub> CreateTpuDriverStub(947 GrpcTpuDriver::CreateTpuDriverStub( in CreateTpuDriverStub() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver990 auto stub = CreateTpuDriverStub(config_, creds_); in AllocateStream()998 auto stub = CreateTpuDriverStub(config_, creds_); in QuerySystemInfo()1016 auto stub = CreateTpuDriverStub(config_, creds_); in Reset()1038 auto stub = CreateTpuDriverStub(config_, creds_); in Close()1060 auto stub = GrpcTpuDriver::CreateTpuDriverStub(config, creds); in CreateGrpcTpuDriver()