(x1, x2)
| 284 | from torch._dynamo.utils import same |
| 285 | |
| 286 | def foo2(x1, x2): |
| 287 | a = torch.neg(x1) |
| 288 | b = torch.maximum(x2, a) |
| 289 | y = torch.cat([b], dim=0) |
| 290 | return y |
| 291 | |
| 292 | x1 = torch.randn((1, 8), dtype=torch.float32) |
| 293 | x2 = torch.randn((8390, 8), dtype=torch.float32) |