Searched refs:DirectTpuDriver (Results 1 – 1 of 1) sorted by relevance
60 class DirectTpuDriver;112 friend DirectTpuDriver;134 friend DirectTpuDriver;171 friend DirectTpuDriver;190 friend DirectTpuDriver;236 class DirectTpuDriver : public TpuDriver { class238 explicit DirectTpuDriver(const std::string& so_path) { in DirectTpuDriver() function in tpu_driver::__anon9d833a900111::DirectTpuDriver254 DirectTpuDriver() { in DirectTpuDriver() function in tpu_driver::__anon9d833a900111::DirectTpuDriver260 ~DirectTpuDriver() override { driver_fn_.TpuDriver_Close(driver_); } in ~DirectTpuDriver()490 absl::make_unique<DirectTpuDriver>()); in RegisterDirectTpuDriver()[all …]