Searched defs:sticky_device (Results 1 – 3 of 3) sorted by relevance
141 Status set_sticky_device(PjRtDevice* sticky_device) { in set_sticky_device()147 PjRtDevice* sticky_device() const { return sticky_device_; } in sticky_device() function
567 [](PyBuffer::object self, PjRtDevice* sticky_device) { in RegisterTypes()
343 std::optional<xla::ClientAndPtr<xla::PjRtDevice>> sticky_device; member