Home
last modified time | relevance | path

Searched refs:TpuDriver_TransferFromDevice (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu.h246 TpuDriver_TransferFromDevice;
288 TpuDriver_TransferFromDevice; // NOLINT member
Dlibtpu_client.c131 driver_fn.TpuDriver_TransferFromDevice(driver, buf_sum_handle, sum_src, in main()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc357 driver_fn_.TpuDriver_TransferFromDevice( in TransferFromDevice()