Searched refs:always_fallback_to_python_ (Results 1 – 2 of 2) sorted by relevance
325 bool always_fallback_to_python_ = false; member in jax::PmapFunction370 always_fallback_to_python_ = true; in PopulateCacheEntry()414 if (always_fallback_to_python_) { in Call()
536 bool always_fallback_to_python_ = false; member in jax::__anon838f46370811::CompiledFunction827 always_fallback_to_python_ = true; in TryToPopulateDefaultDevice()841 always_fallback_to_python_ = true; in TryToPopulateDefaultDevice()863 if (always_fallback_to_python_) { in Call()