(self, device)
| 578 | return qs_packed_dict |
| 579 | |
| 580 | def to(self, device): |
| 581 | # make sure the quantization state is on the right device |
| 582 | self.code = self.code.to(device) |
| 583 | self.absmax = self.absmax.to(device) |
| 584 | if self.nested: |
| 585 | self.offset = self.offset.to(device) |
| 586 | self.state2.absmax = self.state2.absmax.to(device) |
| 587 | self.state2.code = self.state2.code.to(device) |
| 588 | |
| 589 | def __eq__(self, other): |
| 590 | if not isinstance(other, QuantState): |
no outgoing calls