Searched refs:get_jax_enable_x64_ (Results 1 – 2 of 2) sorted by relevance
139 get_jax_enable_x64_(get_jax_enable_x64) { in PmapFunction()185 const py::function get_jax_enable_x64_; member in jax::PmapFunction261 jax_enable_x64_ = py::cast<bool>(get_jax_enable_x64_()); in Call()
892 const py::function get_jax_enable_x64_; member in jax::__anone04173081411::CompiledFunction924 get_jax_enable_x64_(get_jax_enable_x64), in CompiledFunction()1118 jax_enable_x64_ = py::cast<bool>(get_jax_enable_x64_()); in Call()