Home
last modified time | relevance | path

Searched defs:TransferToDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc371 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon0a9a835f0111::GrpcTpuDriver
783 std::shared_ptr<Event> GrpcTpuStream::TransferToDevice( in TransferToDevice() function in tpu_driver::__anon0a9a835f0111::GrpcTpuStream
Ddirect_tpu_driver.cc338 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon8b72a3ae0111::DirectTpuDriver
Drecording_tpu_driver.cc266 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anonfb68d0100111::RecordingTpuDriver
Dpod_tpu_driver.cc463 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon238e18760111::PodTpuDriver