Searched defs:has_explicit_device (Results 1 – 1 of 1) sorted by relevance
508 bool has_explicit_device() const { return has_explicit_device_; } in has_explicit_device() function in jax::__anon901e1bb70811::CompiledFunction607 bool has_explicit_device, in CompiledFunction()1202 bool has_explicit_device, in InitializeCompiledFunction()1217 bool has_explicit_device, in MakeCompiledFunction()1358 bool has_explicit_device = in BuildJaxjitSubmodule() local1401 std::shared_ptr<CompiledFunctionCache> cache) -> py::object { in BuildJaxjitSubmodule()