Searched defs:cache_miss (Results 1 – 2 of 2) sorted by relevance
134 PmapFunction(py::function fun, py::function cache_miss, in PmapFunction()426 std::vector<int> static_argnums) -> std::unique_ptr<PmapFunction> { in BuildPmapSubmodule()
916 CompiledFunction::CompiledFunction(py::function fun, py::function cache_miss, in CompiledFunction()1224 std::vector<int> static_argnums) -> std::unique_ptr<CompiledFunction> { in BuildJaxjitSubmodule()