Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Dpod_tpu_driver.cc188 class PodCompiledProgramHandle : public CompiledProgramHandle { class
190 explicit PodCompiledProgramHandle(PodTpuDriver* driver, int64_t operation_id) in PodCompiledProgramHandle() function in tpu_driver::__anonf243d9070111::PodCompiledProgramHandle
589 return std::make_unique<PodCompiledProgramHandle>(this, operation_id); in CompileProgram()
598 static_cast<const PodCompiledProgramHandle*>(handle)->operation_id()); in LoadProgram()
600 static_cast<const PodCompiledProgramHandle*>(handle)->operation_id(); in LoadProgram()
966 xla::Status PodCompiledProgramHandle::program_shape( in program_shape()