(self, numel: int, numel_padded: int, dtype: torch.dtype, source: Optional[torch.Tensor] = None)
| 30 | """ |
| 31 | |
| 32 | def __init__(self, numel: int, numel_padded: int, dtype: torch.dtype, source: Optional[torch.Tensor] = None): |
| 33 | self.numel = numel |
| 34 | self.numel_padded = numel_padded |
| 35 | self.dtype = dtype |
| 36 | if source is not None: |
| 37 | self.data = source |
| 38 | else: |
| 39 | self.data = torch.zeros(self.numel_padded, dtype=self.dtype, device=get_device_name(), requires_grad=False) |
| 40 | |
| 41 | def zero(self): |
| 42 | """Reset the buffer to zero.""" |
no test coverage detected