Searched defs:compile_function (Results 1 – 3 of 3) sorted by relevance
370 const std::function<Status(TpuProgramGroupInterface*)>& compile_function) { in CompileIfKeyAbsent()409 const std::function<Status(TpuProgramGroupInterface*)>& compile_function) { in CompileIfKeyAbsentHelper()
216 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>& in CompileIfKeyAbsent()
377 def compile_function( function