MCPcopy Create free account
hub / github.com/BICLab/SpikingBrain-7B / forward

Method forward

W8ASpike/activations.py:135–140  ·  view source on GitHub ↗
(ctx, x)

Source from the content-addressed store, hash-verified

133 @staticmethod
134 @contiguous
135 def forward(ctx, x):
136 T, D = x.numel(), x.shape[-1]
137 y = torch.empty_like(x)
138 logsigmoid_fwd_kernel[lambda meta: (triton.cdiv(meta['T'], meta['D']),)](x, y, T=T, D=D)
139 ctx.save_for_backward(x,)
140 return y
141
142 @staticmethod
143 @contiguous

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected