xref: /aosp_15_r20/external/pytorch/test/package/test_trace_dep/__init__.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1import yaml
2
3import torch
4
5
6class SumMod(torch.nn.Module):
7    def forward(self, inp):
8        return torch.sum(inp)
9