MCPcopy Create free account
hub / github.com/HazyResearch/spacetime / backward

Method backward

model/functional/toeplitz.py:65–69  ·  view source on GitHub ↗
(ctx, grad)

Source from the content-addressed store, hash-verified

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
71class TriangularToeplitzMultFast(torch.autograd.Function):
72 @staticmethod

Callers 1

toeplitz.pyFile · 0.45

Calls 1

Tested by

no test coverage detected