Skip to content

Commit 24f8528

Browse files
committed
change to relu
1 parent 34223a0 commit 24f8528

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

backends/qualcomm/tests/models.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,11 @@
1414
class Alias(torch.nn.Module):
1515
def __init__(self):
1616
super().__init__()
17-
self.fc = torch.nn.Linear(10, 10)
17+
self.relu = torch.nn.ReLU()
1818

1919
def forward(self, x):
2020
alias_x = torch.ops.aten.alias.default(x)
21-
out = self.fc(alias_x)
22-
return out
21+
return self.relu(alias_x)
2322

2423

2524
class And(torch.nn.Module):

0 commit comments

Comments
 (0)