mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
b0c27b44cf
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57498 Test Plan: Imported from OSS Reviewed By: albanD Differential Revision: D28162448 Pulled By: tugsbayasgalan fbshipit-source-id: 5c21ced42a22aca7cee089e876e9d98d32f68955
10 lines
224 B
Text
10 lines
224 B
Text
def while_test(a: Tensor,
|
|
i: Tensor) -> Tensor:
|
|
a0 = a
|
|
i0 = i
|
|
_0 = bool(torch.lt(i, 3))
|
|
while _0:
|
|
a1 = torch.mul_(a0, a0)
|
|
i1 = torch.add_(i0, 1)
|
|
_0, a0, i0 = bool(torch.lt(i1, 3)), a1, i1
|
|
return a0
|