Searched defs:disable_jit (Results 1 – 2 of 2) sorted by relevance
43 bool disable_jit = false; member65 absl::optional<bool> disable_jit; member
1299 [&](bool disable_jit) { global_state.disable_jit = disable_jit; }); in BuildJaxjitSubmodule()1303 [&](absl::optional<bool> disable_jit) { in BuildJaxjitSubmodule()1309 jitlib.def("set_disable_jit", [&](bool disable_jit) { in BuildJaxjitSubmodule()