Searched refs:InitializePmapFunction (Results 1 – 1 of 1) sorted by relevance
708 void InitializePmapFunction(JaxPmapFunctionObject* cfun, py::function fun, in InitializePmapFunction() function726 InitializePmapFunction(buf, std::move(fun), std::move(cache_miss), in MakePmapFunction()906 InitializePmapFunction( in BuildPmapSubmodule()