Searched refs:AsCompiledFunction (Results 1 – 1 of 1) sorted by relevance
1052 xla::StatusOr<CompiledFunction*> AsCompiledFunction(py::handle handle) { in AsCompiledFunction() function1321 TF_ASSIGN_OR_RETURN(CompiledFunction * fun, AsCompiledFunction(self)); in BuildJaxjitSubmodule()1326 TF_ASSIGN_OR_RETURN(CompiledFunction * fun, AsCompiledFunction(self)); in BuildJaxjitSubmodule()1468 TF_ASSIGN_OR_RETURN(CompiledFunction * fun, AsCompiledFunction(self)); in BuildJaxjitSubmodule()1474 TF_ASSIGN_OR_RETURN(CompiledFunction * fun, AsCompiledFunction(self)); in BuildJaxjitSubmodule()