Searched refs:PopulateAndSaveEntry (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | recording_tpu_driver.cc | 175 PopulateAndSaveEntry(&r, wait_for, handle_id, thread_id); in Allocate()
198 PopulateAndSaveEntry(&r, wait_for, handle_id, thread_id); in Allocate()
236 PopulateAndSaveEntry(&r, wait_for, handle_id, thread_id); in AllocateTuple()
258 PopulateAndSaveEntry(&r, wait_for, event_id, thread_id); in Deallocate()
288 PopulateAndSaveEntry(&r, wait_for, event_id, thread_id); in TransferToDevice()
310 PopulateAndSaveEntry(&r, wait_for, event_id, thread_id); in TransferFromDevice()
335 PopulateAndSaveEntry(&r, wait_for, event_id, thread_id); in TransferFromDeviceToDevice()
355 PopulateAndSaveEntry(&r, wait_for, handle_id, thread_id); in CompileProgram()
380 PopulateAndSaveEntry(&r, wait_for, handle_id, thread_id); in LoadProgram()
404 PopulateAndSaveEntry(&r, wait_for, event_id, thread_id); in UnloadProgram()
[all …]
|