Home
last modified time | relevance | path

Searched defs:pjrt_executable (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.h127 const PjRtLoadedExecutable& pjrt_executable() const { return *executable_; } in pjrt_executable() function
H A Dpmap_lib.cc511 auto pjrt_executable = cache_entry.executable->mutable_pjrt_executable(); in Call() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc200 auto pjrt_executable = std::make_unique<PjRtStreamExecutorExecutable>( in DeserializeExecutable() local