Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.h206 virtual std::shared_ptr<Event> TransferToDevice(
Drecording_tpu_driver.cc264 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anone63593b20111::RecordingTpuDriver
274 std::make_shared<RecordingEvent>(driver_->TransferToDevice( in TransferToDevice()
Dgrpc_tpu_driver.cc189 std::shared_ptr<Event> TransferToDevice(const void* src, BufferHandle* dst,
371 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver
375 return stream->TransferToDevice(src, dst, wait_for); in TransferToDevice()
783 std::shared_ptr<Event> GrpcTpuStream::TransferToDevice( in TransferToDevice() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
Dpod_tpu_driver.cc463 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon6b014a980111::PodTpuDriver
480 return core_to_driver_[core_id]->TransferToDevice( in TransferToDevice()
553 auto dst_event = TransferToDevice(host_buf, dst, {src_event.get()}); in TransferFromDeviceToDevice()
Ddirect_tpu_driver.cc338 std::shared_ptr<Event> TransferToDevice( in TransferToDevice() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc191 driver->TransferToDevice(leaves[0].untyped_data(), handle, {}); in FromLiterals()
213 return driver->TransferToDevice(leaf.untyped_data(), handle, {}); in FromLiterals()