Searched refs:test_foreach_add (Results 1 – 1 of 1) sorted by relevance
743 def test_foreach_add(a0, a1, b0, b1): function753 out_eager = test_foreach_add(*inps)754 out_compiled = torch.compile(test_foreach_add)(*inps)761 def test_foreach_add(a0, a1, a2, b0, b1, b2): function775 out_eager = test_foreach_add(*inps)776 out_compiled = torch.compile(test_foreach_add)(*inps)