Searched defs:cfun (Results 1 – 2 of 2) sorted by relevance
413 py::class_<PmapFunction, std::unique_ptr<PmapFunction>> cfun(pmap_lib, in BuildPmapSubmodule() local
1087 void InitializeCompiledFunction(JaxCompiledFunctionObject* cfun, in InitializeCompiledFunction()1181 py::object cfun; in BuildJaxjitSubmodule() local