(module, name, buffer, persistent=True)
| 19 | ) |
| 20 | |
| 21 | def register_empty_buffer(module, name, buffer, persistent=True): |
| 22 | old_register_buffer(module, name, buffer, persistent=persistent) |
| 23 | if buffer is not None: |
| 24 | module._buffers[name] = module._buffers[name].to(device) |
| 25 | |
| 26 | def patch_tensor_constructor(fn): |
| 27 | def wrapper(*args, **kwargs): |
nothing calls this directly
no outgoing calls
no test coverage detected