Forward pass of the BitFeedForward module. Args: x (torch.Tensor): The input tensor. Returns: torch.Tensor: The output tensor.
(self, x)
| 123 | init_zero_(self.ff[-1]) |
| 124 | |
| 125 | def forward(self, x): |
| 126 | """ |
| 127 | Forward pass of the BitFeedForward module. |
| 128 | |
| 129 | Args: |
| 130 | x (torch.Tensor): The input tensor. |
| 131 | |
| 132 | Returns: |
| 133 | torch.Tensor: The output tensor. |
| 134 | """ |
| 135 | return self.ff(x) |
nothing calls this directly
no outgoing calls
no test coverage detected