| 56 | return output |
| 57 | |
| 58 | class TriangularToeplitzMult(torch.autograd.Function): |
| 59 | @staticmethod |
| 60 | def forward(ctx, u, v): |
| 61 | ctx.save_for_backward(u, v) |
| 62 | return triangular_toeplitz_multiply_(u, v) |
| 63 | |
| 64 | @staticmethod |
| 65 | def backward(ctx, grad): |
| 66 | u, v = ctx.saved_tensors |
| 67 | d_u = triangular_toeplitz_multiply_(grad.flip(-1), v).flip(-1) |
| 68 | d_v = triangular_toeplitz_multiply_(grad.flip(-1), u).flip(-1) |
| 69 | return d_u, d_v |
| 70 | |
| 71 | class TriangularToeplitzMultFast(torch.autograd.Function): |
| 72 | @staticmethod |
nothing calls this directly
no outgoing calls
no test coverage detected