Searched refs:CreateGrpcTpuDriver (Results 1 – 3 of 3) sorted by relevance
25 xla::StatusOr<std::unique_ptr<TpuDriver>> CreateGrpcTpuDriver(
1057 xla::StatusOr<std::unique_ptr<TpuDriver>> CreateGrpcTpuDriver( in CreateGrpcTpuDriver() function1090 return CreateGrpcTpuDriver( in __anon4073f9c10402()1093 return CreateGrpcTpuDriver(config, in __anon4073f9c10402()
264 CreateGrpcTpuDriver(worker_config, creds_).ConsumeValueOrDie(); in PodTpuDriver()