Searched refs:SetStickyDevice (Results 1 – 4 of 4) sorted by relevance
104 void SetStickyDevice(pybind11::object sticky_device);
326 void PyBuffer::SetStickyDevice(pybind11::object sticky_device) { in SetStickyDevice() function in xla::PyBuffer
278 &PyBuffer::SetStickyDevice) in PYBIND11_MODULE()
1192 buffer->SetStickyDevice(sticky_device); in Call()1253 buffer->SetStickyDevice(py::none()); in BuildJaxjitSubmodule()