Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc377 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anonbd0c70900111::GrpcTpuDriver
798 std::shared_ptr<Event> GrpcTpuStream::TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anonbd0c70900111::GrpcTpuStream
Ddirect_tpu_driver.cc351 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anona214a11f0111::DirectTpuDriver
Drecording_tpu_driver.cc296 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon2563a3e10111::RecordingTpuDriver
Dpod_tpu_driver.cc488 std::shared_ptr<Event> TransferFromDevice( in TransferFromDevice() function in tpu_driver::__anon09eea3870111::PodTpuDriver