Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Dpod_tpu_driver.cc207 class PodLoadedProgramHandle : public LoadedProgramHandle { class
209 explicit PodLoadedProgramHandle(PodTpuDriver* driver, int64_t operation_id, in PodLoadedProgramHandle() function in tpu_driver::__anonf243d9070111::PodLoadedProgramHandle
620 return std::make_unique<PodLoadedProgramHandle>(this, operation_id, in LoadProgram()
630 static_cast<PodLoadedProgramHandle*>(handle.get())->operation_id()); in UnloadProgram()
632 static_cast<PodLoadedProgramHandle*>(handle.get())->operation_id(); in UnloadProgram()
634 static_cast<PodLoadedProgramHandle*>(handle.get())->core_id(); in UnloadProgram()
661 deps.insert(static_cast<PodLoadedProgramHandle*>(program)->operation_id()); in ExecuteProgram()
663 auto op_id = static_cast<PodLoadedProgramHandle*>(program)->operation_id(); in ExecuteProgram()
664 auto core_id = static_cast<PodLoadedProgramHandle*>(program)->core_id(); in ExecuteProgram()