Method forward

backends/mlx/test/test_ops.py:49–52

(self, x: torch.Tensor, y: torch.Tensor)

        self.alpha = alpha

    def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
        if self.alpha is not None:
            return torch.add(x, y, alpha=self.alpha)
        return x + y


class AddScalarModel(nn.Module):