Searched defs:cfun (Results 1 – 2 of 2) sorted by relevance
708 void InitializePmapFunction(JaxPmapFunctionObject* cfun, py::function fun, in InitializePmapFunction()830 py::object cfun; in BuildPmapSubmodule() local
1199 void InitializeCompiledFunction(JaxCompiledFunctionObject* cfun, in InitializeCompiledFunction()1281 py::object cfun; in BuildJaxjitSubmodule() local