Home
last modified time | relevance | path

Searched defs:tpu_events (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc278 auto tpu_events = MakeEventArray(wait_for); in Allocate() local
290 auto tpu_events = MakeEventArray(wait_for); in Allocate() local
307 auto tpu_events = MakeEventArray(wait_for); in AllocateTuple() local
328 auto tpu_events = MakeEventArray(wait_for); in Deallocate() local
341 auto tpu_events = MakeEventArray(wait_for); in TransferToDevice() local
354 auto tpu_events = MakeEventArray(wait_for); in TransferFromDevice() local
367 auto tpu_events = MakeEventArray(wait_for); in TransferFromDeviceToDevice() local
381 auto tpu_events = MakeEventArray(wait_for); in CompileProgram() local
403 auto tpu_events = MakeEventArray(wait_for); in LoadProgram() local
419 auto tpu_events = MakeEventArray(wait_for); in UnloadProgram() local
[all …]