Skip to content

Commit 89910fe

Browse files
committed
up
1 parent d374c89 commit 89910fe

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/apple/coreml/test/test_torch_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def _coreml_partitioner(self):
3535

3636
def _get_test_model(self):
3737
model = torch.nn.Sequential(
38-
torch.nn.Embedding(64, 128), torch.nn.Linear(128, 128), torch.nn.ReLU()
38+
torch.nn.Embedding(64, 128), torch.nn.Linear(128, 256), torch.nn.ReLU()
3939
)
4040
example_inputs = (torch.LongTensor([0]),)
4141
return model, example_inputs

third-party/ao

Submodule ao updated 62 files

0 commit comments

Comments
 (0)