(self, hidden_states: Optional[jt.Var])
| 188 | self.dropout = nn.Dropout(config.resid_pdrop) |
| 189 | |
| 190 | def execute(self, hidden_states: Optional[jt.Var]) -> jt.Var: |
| 191 | hidden_states = self.fc_in(hidden_states) |
| 192 | hidden_states = self.act(hidden_states) |
| 193 | hidden_states = self.fc_out(hidden_states) |
| 194 | hidden_states = self.dropout(hidden_states) |
| 195 | return hidden_states |
| 196 | |
| 197 | |
| 198 | class MossBlock(Module): |
nothing calls this directly
no outgoing calls
no test coverage detected