Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu.h71 typedef struct TpuStatus { struct
74 } TpuStatus; argument
77 struct TpuStatus* status;
96 typedef struct TpuStatus*(PrototypeTpuDriver_Reset)(struct TpuDriver* driver);
109 typedef struct TpuStatus*(PrototypeTpuDriver_LinearizeShape)(
113 typedef struct TpuStatus*(PrototypeTpuDriver_DelinearizeShape)(
199 void (*callback_fn)(struct TpuStatus*, void* additional_info),
202 typedef struct TpuStatus*(PrototypeTpuDriver_EventAwait)(struct TpuEvent* event,
207 typedef void(PrototypeTpuDriver_FreeStatus)(struct TpuStatus* status);
Dlibtpu_client.c134 TpuStatus* status = driver_fn.TpuDriver_EventAwait(transfer_sum_event, in main()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc36 xla::Status CreateXlaStatus(::TpuStatus* status) { in CreateXlaStatus()
98 [](struct TpuStatus* status, void* additional_info) { in AddCallback()