Searched refs:jax_disable_jit_ (Results 1 – 1 of 1) sorted by relevance
872 bool JitIsDisabled() { return GetDisableJit() || jax_disable_jit_.value(); } in JitIsDisabled()901 absl::optional<bool> jax_disable_jit_ = absl::nullopt; member in jax::__anone04173081411::CompiledFunction1119 jax_disable_jit_ = py::cast<bool>(get_jax_disable_jit_()); in Call()