Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc174 class DirectLoadedProgramHandle : public LoadedProgramHandle { class
176 explicit DirectLoadedProgramHandle(::TpuDriverFn* driver_fn, in DirectLoadedProgramHandle() function in tpu_driver::__anon9d833a900111::DirectLoadedProgramHandle
405 auto loaded_handle = absl::make_unique<DirectLoadedProgramHandle>( in LoadProgram()
420 auto* direct_lph = static_cast<DirectLoadedProgramHandle*>(handle.get()); in UnloadProgram()
457 driver_, static_cast<DirectLoadedProgramHandle*>(program)->handle_, in ExecuteProgram()