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

Method forward

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

Source from the content-addressed store, hash-verified

17
18class SparseSiLU(nn.SiLU):
19 def forward(self, input: SparseTensor) -> SparseTensor:
20 return input.replace(super().forward(input.feats))
21
22
23class SparseGELU(nn.GELU):

Callers

nothing calls this directly

Calls 2

replaceMethod · 0.80
forwardMethod · 0.45

Tested by

no test coverage detected