Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Drecording_tpu_driver.cc109 class RecordingLoadedProgramHandle : public LoadedProgramHandle { class
111 explicit RecordingLoadedProgramHandle( in RecordingLoadedProgramHandle() function in tpu_driver::__anon852af9610111::RecordingLoadedProgramHandle
372 std::make_unique<RecordingLoadedProgramHandle>(driver_->LoadProgram( in LoadProgram()
395 static_cast<RecordingLoadedProgramHandle*>(handle.get())->id_; in UnloadProgram()
398 std::move(static_cast<RecordingLoadedProgramHandle*>(handle.get()) in UnloadProgram()
421 static_cast<RecordingLoadedProgramHandle*>(program)->id_; in ExecuteProgram()
451 static_cast<RecordingLoadedProgramHandle*>(program)->handle_.get(), in ExecuteProgram()