(self, device=torch.device("cpu"))
| 26 | self._decode = {device: self.key2xyz(r512, 9)} |
| 27 | |
| 28 | def encode_lut(self, device=torch.device("cpu")): |
| 29 | if device not in self._encode: |
| 30 | cpu = torch.device("cpu") |
| 31 | self._encode[device] = tuple(e.to(device) for e in self._encode[cpu]) |
| 32 | return self._encode[device] |
| 33 | |
| 34 | def decode_lut(self, device=torch.device("cpu")): |
| 35 | if device not in self._decode: |