Searched defs:get_jax_disable_jit (Results 1 – 1 of 1) sorted by relevance
919 py::function get_jax_disable_jit, in CompiledFunction()1224 std::vector<int> static_argnums) -> std::unique_ptr<CompiledFunction> { in BuildJaxjitSubmodule()