Searched defs:TransferToDevice (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ | ||
D | grpc_tpu_driver.cc | 371 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver |
D | direct_tpu_driver.cc | 338 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon9d833a900111::DirectTpuDriver |
D | recording_tpu_driver.cc | 264 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anone63593b20111::RecordingTpuDriver |
D | pod_tpu_driver.cc | 463 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon6b014a980111::PodTpuDriver |