Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dlibtpu_client.c29 struct TpuDriverFn* driver_fn) { in LoadAndInitializeDriver() argument
39 initialize_fn(driver_fn, true); in LoadAndInitializeDriver()
50 struct TpuDriverFn driver_fn; in main() local
51 void* handle = LoadAndInitializeDriver(api_path, &driver_fn); in main()
54 fprintf(stdout, "TPU Driver Version: %s\n", driver_fn.TpuDriver_Version()); in main()
57 struct TpuDriver* driver = driver_fn.TpuDriver_Open("local://"); in main()
60 struct TpuSystemInfo* info = driver_fn.TpuDriver_QuerySystemInfo(driver); in main()
61 driver_fn.TpuDriver_FreeSystemInfo(info); in main()
74 driver_fn.TpuDriver_CompileProgramFromText(driver, hlo_module_text, in main()
80 driver_fn.TpuDriver_LoadProgram(driver, /*core_id=*/0, cph, in main()
[all …]
Dlibtpu.h92 typedef void(PrototypeTpuDriver_Initialize)(struct TpuDriverFn* driver_fn,
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc64 explicit DirectEvent(::TpuDriverFn* driver_fn, ::TpuEvent* event) in DirectEvent() argument
65 : driver_fn_(driver_fn), event_(event) {} in DirectEvent()
117 explicit DirectBufferHandle(::TpuDriverFn* driver_fn, in DirectBufferHandle() argument
119 : handle_(handle), event_(new DirectEvent(driver_fn, handle->event)) {} in DirectBufferHandle()
139 explicit DirectCompiledProgramHandle(::TpuDriverFn* driver_fn, in DirectCompiledProgramHandle() argument
142 driver_fn_(driver_fn), in DirectCompiledProgramHandle()
143 event_(new DirectEvent(driver_fn, handle->event)) {} in DirectCompiledProgramHandle()
176 explicit DirectLoadedProgramHandle(::TpuDriverFn* driver_fn, in DirectLoadedProgramHandle() argument
178 : handle_(handle), event_(new DirectEvent(driver_fn, handle->event)) {} in DirectLoadedProgramHandle()
195 explicit DirectTpuLinearizer(::TpuDriver* driver, ::TpuDriverFn* driver_fn) in DirectTpuLinearizer() argument
[all …]