Searched refs:always_fallback_to_python_ (Results 1 – 2 of 2) sorted by relevance
172 bool always_fallback_to_python_ = false; member in jax::PmapFunction256 if (always_fallback_to_python_) { in Call()
874 bool always_fallback_to_python_ = false; member in jax::__anone04173081411::CompiledFunction1106 if (always_fallback_to_python_) { in Call()1127 always_fallback_to_python_ = true; in Call()1133 always_fallback_to_python_ = true; in Call()