(self, x, y, z, xpos, ypos)
| 380 | self.norm_z = norm_layer(dim) if norm_mem else nn.Identity() |
| 381 | |
| 382 | def forward(self, x, y, z, xpos, ypos): |
| 383 | x = x + self.drop_path(self.attn(self.norm1(x), xpos)) |
| 384 | y_ = self.norm_y(y) |
| 385 | z_ = self.norm_z(z) |
| 386 | x = x + self.drop_path(self.cross_attn(self.norm2(x), y_, z_, xpos, ypos)) |
| 387 | x = x + self.drop_path(self.mlp(self.norm3(x))) |
| 388 | return x, y |
| 389 | |
| 390 | |
| 391 | class ModLN(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected