Searched refs:BuildJaxjitSubmodule (Results 1 – 3 of 3) sorted by relevance
174 void BuildJaxjitSubmodule(pybind11::module& m);
372 jax::BuildJaxjitSubmodule(m); in PYBIND11_MODULE()
1209 void BuildJaxjitSubmodule(pybind11::module& m) { in BuildJaxjitSubmodule() function