Home
last modified time | relevance | path

Searched refs:TransferFromDeviceToDevice (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.h213 virtual std::shared_ptr<Event> TransferFromDeviceToDevice(
Drecording_tpu_driver.cc316 std::shared_ptr<Event> TransferFromDeviceToDevice( in TransferFromDeviceToDevice() function in tpu_driver::__anone63593b20111::RecordingTpuDriver
325 std::make_shared<RecordingEvent>(driver_->TransferFromDeviceToDevice( in TransferFromDeviceToDevice()
Dgrpc_tpu_driver.cc194 std::shared_ptr<Event> TransferFromDeviceToDevice(
384 std::shared_ptr<Event> TransferFromDeviceToDevice( in TransferFromDeviceToDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver
388 return stream->TransferFromDeviceToDevice(src, dst, wait_for); in TransferFromDeviceToDevice()
816 std::shared_ptr<Event> GrpcTpuStream::TransferFromDeviceToDevice( in TransferFromDeviceToDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
Dpod_tpu_driver.cc512 std::shared_ptr<Event> TransferFromDeviceToDevice( in TransferFromDeviceToDevice() function in tpu_driver::__anon6b014a980111::PodTpuDriver
542 return core_to_driver_[dst_core_id]->TransferFromDeviceToDevice( in TransferFromDeviceToDevice()
Ddirect_tpu_driver.cc364 std::shared_ptr<Event> TransferFromDeviceToDevice( in TransferFromDeviceToDevice() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc414 return driver->TransferFromDeviceToDevice( in CopyToDevice()