Searched refs:BuildJaxjitSubmodule (Results 1 – 3 of 3) sorted by relevance
163 void BuildJaxjitSubmodule(pybind11::module& m);
416 jax::BuildJaxjitSubmodule(m); in PYBIND11_MODULE()
1245 void BuildJaxjitSubmodule(py::module& m) { in BuildJaxjitSubmodule() function