Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Dpod_tpu_driver.cc799 xla::Status GetCompiledProgramShape(int64_t op_id, in GetCompiledProgramShape() function in tpu_driver::__anonf243d9070111::PodTpuDriver
968 return driver_->GetCompiledProgramShape(operation_id(), program_shape); in program_shape()