Home
last modified time | relevance | path

Searched refs:always_fallback_to_python_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpmap_lib.cc325 bool always_fallback_to_python_ = false; member in jax::PmapFunction
370 always_fallback_to_python_ = true; in PopulateCacheEntry()
414 if (always_fallback_to_python_) { in Call()
H A Djax_jit.cc536 bool always_fallback_to_python_ = false; member in jax::__anon901e1bb70811::CompiledFunction
827 always_fallback_to_python_ = true; in TryToPopulateDefaultDevice()
841 always_fallback_to_python_ = true; in TryToPopulateDefaultDevice()
863 if (always_fallback_to_python_) { in Call()