Home
last modified time | relevance | path

Searched refs:tpu_driver (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
DBUILD5 "//tensorflow/compiler/xla/python/tpu_driver:platform/external/tools.bzl",
16 srcs = ["tpu_driver.proto"],
36 name = "tpu_driver",
38 "tpu_driver.cc",
43 "tpu_driver.h",
64 ":tpu_driver",
82 ":tpu_driver",
92 "//tensorflow/compiler/xla/python/tpu_driver/client:libtpu",
103 ":tpu_driver",
124 ":tpu_driver",
Dtpu_driver.h42 namespace tpu_driver {
123 return ::tpu_driver::ComputeBytesFromShape(shape); in ComputeBytesFromShape()
249 ::tpu_driver::TpuDriverRegistry::RegisterDriver(prefix, fn);
Dtpu_service.proto18 package tpu_driver; package
20 import "tensorflow/compiler/xla/python/tpu_driver/tpu_driver.proto";
Dgrpc_tpu_driver.h23 namespace tpu_driver {
Devent_id.h25 namespace tpu_driver {
Dtpu_driver.cc22 namespace tpu_driver { namespace
Dtpu_driver.proto18 package tpu_driver; package
Dpod_tpu_driver.cc30 namespace tpu_driver { namespace
263 auto tpu_driver = in PodTpuDriver() local
267 tpu_driver->QuerySystemInfo(&driver_info); in PodTpuDriver()
270 drivers_.insert({worker_count, std::move(tpu_driver)}); in PodTpuDriver()
273 drivers_.insert({driver_info.host_id(), std::move(tpu_driver)}); in PodTpuDriver()
Ddirect_tpu_driver.cc27 namespace tpu_driver { namespace
Drecording_tpu_driver.cc36 namespace tpu_driver { namespace
Dgrpc_tpu_driver.cc35 namespace tpu_driver { namespace
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h54 GetTpuDevices(const tpu_driver::SystemInfo& system_info);
93 std::unique_ptr<tpu_driver::TpuDriver> driver,
129 tpu_driver::TpuDriver* driver() { return driver_.get(); } in driver()
135 std::unique_ptr<tpu_driver::TpuDriver> driver_;
153 TpuSharedBuffer(tpu_driver::TpuDriver* driver, in TpuSharedBuffer()
154 std::unique_ptr<tpu_driver::BufferHandle> handle, in TpuSharedBuffer()
155 std::vector<std::shared_ptr<tpu_driver::Event>> wait_for_use, in TpuSharedBuffer()
163 std::vector<tpu_driver::Event*> events; in ~TpuSharedBuffer()
170 tpu_driver::TpuDriver* const driver;
173 std::unique_ptr<tpu_driver::BufferHandle> handle;
[all …]
Dtpu_client.cc51 TpuDevice::GetTpuDevices(const tpu_driver::SystemInfo& system_info) { in GetTpuDevices()
69 tpu_driver::TpuDriverConfig driver_config; in Get()
71 auto client_status = tpu_driver::TpuDriverRegistry::Open(driver_config); in Get()
78 tpu_driver::SystemInfo system_info; in Get()
90 std::unique_ptr<tpu_driver::TpuDriver> driver, in PyTpuClient()
182 tpu_driver::TpuDriver* driver = client->driver(); in FromLiterals()
189 leaves_references](tpu_driver::BufferHandle* handle) { in FromLiterals()
212 [driver, &leaf, &indexed_shape](tpu_driver::BufferHandle* handle) { in FromLiterals()
235 std::vector<tpu_driver::BufferHandle*> child_handle_ptrs; in MakeTuple()
236 std::vector<std::shared_ptr<tpu_driver::Event>> child_events; in MakeTuple()
[all …]
DBUILD32 "//tensorflow/compiler/xla/python/tpu_driver",
33 "//tensorflow/compiler/xla/python/tpu_driver:direct_tpu_driver",
34 "//tensorflow/compiler/xla/python/tpu_driver:grpc_tpu_driver",
35 "//tensorflow/compiler/xla/python/tpu_driver:pod_tpu_driver",
36 "//tensorflow/compiler/xla/python/tpu_driver:recording_tpu_driver",
37 "//tensorflow/compiler/xla/python/tpu_driver:tpu_driver_proto_cc",
Dtpu_client.py26 from tensorflow.compiler.xla.python.tpu_driver.client import tpu_client_extension as _tpu_client
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/platform/external/
Dcompat.h23 namespace tpu_driver {
/external/tensorflow/tensorflow/core/platform/
DBUILD1750 "//learning/brain/tfrc/runtime/tpu_driver:__subpackages__",