Searched refs:default_pydevice_ (Results 1 – 1 of 1) sorted by relevance
911 xla::ClientAndPtr<xla::PjRtDevice> default_pydevice_; member in jax::__anone04173081411::CompiledFunction1123 default_pydevice_ = py::cast<xla::ClientAndPtr<xla::PjRtDevice>>( in Call()1130 default_pyclient_ = default_pydevice_.client; in Call()1131 default_device_ = default_pydevice_.contents; in Call()