Searched refs:sticky_device (Results 1 – 3 of 3) sorted by relevance
326 void PyBuffer::SetStickyDevice(pybind11::object sticky_device) { in SetStickyDevice() argument327 if (sticky_device_ && !sticky_device_->equal(sticky_device)) { in SetStickyDevice()332 sticky_device_ = sticky_device; in SetStickyDevice()
104 void SetStickyDevice(pybind11::object sticky_device);
824 py::object sticky_device; member1084 cache_entry->sticky_device = in AddCacheEntry()1185 const py::object& sticky_device = cache_entry->sticky_device; in Call() local1192 buffer->SetStickyDevice(sticky_device); in Call()1200 (*device_array)(out_avals[i], sticky_device, out_lazy_exprs[i], in Call()