Searched defs:sticky_device (Results 1 – 3 of 3) sorted by relevance
124 Status set_sticky_device(PjRtDevice* sticky_device) { in set_sticky_device()130 PjRtDevice* sticky_device() const { return sticky_device_; } in sticky_device() function
533 [](PyBuffer::object self, PjRtDevice* sticky_device) { in RegisterTypes()
312 xla::PjRtDevice* sticky_device; member