Searched refs:get_device_ (Results 1 – 1 of 1) sorted by relevance
894 const py::function get_device_; member in jax::__anone04173081411::CompiledFunction926 get_device_(std::move(get_device)) { in CompiledFunction()1121 py::object device_and_is_committed = get_device_(); in Call()