MCPcopy Create free account
hub / github.com/microsoft/TRELLIS / forward

Method forward

trellis/modules/sparse/nonlinearity.py:24–25  ·  view source on GitHub ↗
(self, input: SparseTensor)

Source from the content-addressed store, hash-verified

22
23class SparseGELU(nn.GELU):
24 def forward(self, input: SparseTensor) -> SparseTensor:
25 return input.replace(super().forward(input.feats))
26
27
28class SparseActivation(nn.Module):

Callers

nothing calls this directly

Calls 2

replaceMethod · 0.80
forwardMethod · 0.45

Tested by

no test coverage detected