Lines Matching full:ones
170 self.assert_functionalization(f, torch.ones(4))
182 _functionalize(f, reapply_views=True, crossref=self.crossref)(torch.ones(3, 3))
191 torch.ones(2, 2)
197 y = torch.ones(2)
203 r = _functionalize(f, reapply_views=True, crossref=False)(torch.ones(2))
210 val = torch.ones(3, 1)
238 logs = self.get_logs(g, torch.ones(16, 64, 128, 128, requires_grad=True))
277 tmp = torch.ones(4, 2)
283 self.assert_functionalization(f, torch.ones(4, 2))
284 logs = self.get_logs(f, torch.ones(4, 2))
292 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
294 add = torch.ops.aten.add.Tensor(view_copy, ones); view_copy = ones = None
304 f, torch.ones(4, 2), reapply_views=True, run_reinplace=True
313 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
315 add = torch.ops.aten.add.Tensor(view, ones); view = ones = None
326 tmp = torch.ones(4, 2)
334 self.assert_functionalization(f, torch.ones(4, 2))
335 logs = self.get_logs(f, torch.ones(4, 2))
343 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
346 add = torch.ops.aten.add.Tensor(view_copy, ones); view_copy = ones = None
353 f, torch.ones(4, 2), reapply_views=True, run_reinplace=True
362 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
365 add = torch.ops.aten.add.Tensor(view, ones); view = ones = None
475 self.assert_functionalization(f, torch.ones(2, 2), reapply_views=True)
478 nonfunctional_tensor = torch.ones(2, dtype=torch.long)
482 functional_tensor = torch.ones(2, dtype=torch.long)
486 out = f(torch.ones(2, 2))
488 torch.ones(2, 2)
496 tmp = torch.ones(4, 2)
501 self.assert_functionalization(f, torch.ones(4, 2))
502 logs = self.get_logs(f, torch.ones(4, 2))
510 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
512 add = torch.ops.aten.add.Tensor(arg0_1, ones); ones = None
520 f, torch.ones(4, 2), reapply_views=True, run_reinplace=True
529 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
531 add = torch.ops.aten.add.Tensor(arg0_1, ones); ones = None
546 logs = self.get_logs(f, torch.ones(1))
572 self.assert_functionalization(f, torch.ones(9))
573 logs = self.get_logs(f, torch.ones(9))
592 f, torch.ones(2, 2), reapply_views=True, run_reinplace=False
616 self.assert_functionalization(f, torch.ones(2, 2))
617 logs = self.get_logs(f, torch.ones(2, 2))
636 self.assert_functionalization(f, torch.ones(2, 2))
637 logs = self.get_logs(f, torch.ones(2, 2))
652 f, torch.ones(2, 2), reapply_views=True, run_reinplace=True
670 tmp = torch.ones(2)
676 self.assert_functionalization(f, torch.ones(2, 2))
677 logs = self.get_logs(f, torch.ones(2, 2))
685 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
688 add = torch.ops.aten.add.Tensor(diagonal_copy, ones); diagonal_copy = ones = None
697 f, torch.ones(2, 2), reapply_views=True, run_reinplace=True
706 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
709 add = torch.ops.aten.add_.Tensor(diagonal, ones); diagonal = ones = add = None
719 tmp = torch.ones(2)
724 x = torch.ones(2, 2)
726 logs = self.get_logs(f, torch.ones(2, 2))
734 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
736 add = torch.ops.aten.add.Tensor(diagonal_copy, ones); diagonal_copy = ones = None
746 f, torch.ones(2, 2), reapply_views=True, run_reinplace=False
755 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
757 add = torch.ops.aten.add.Tensor(diagonal, ones); diagonal = ones = None
768 tmp = torch.ones(2)
787 tmp = torch.ones(2)
794 self.assert_functionalization(f, torch.ones(4, 2))
795 logs = self.get_logs(f, torch.ones(4, 2))
803 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
808 add = torch.ops.aten.add.Tensor(diagonal_copy, ones); diagonal_copy = ones = None
826 f, torch.ones(4, 2), reapply_views=True, run_reinplace=False
835 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
840 add = torch.ops.aten.add.Tensor(diagonal, ones); diagonal = ones = None
859 tmp = torch.ones(2)
866 self.assert_functionalization(f, torch.ones(4, 2))
867 logs = self.get_logs(f, torch.ones(4, 2))
875 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
880 add = torch.ops.aten.add.Tensor(diagonal_copy, ones); diagonal_copy = ones = None
898 f, torch.ones(4, 2), reapply_views=True, run_reinplace=False
907 ones = torch.ops.aten.ones.default([2], device = device(type='cpu'), pin_memory = False)
912 add = torch.ops.aten.add.Tensor(diagonal, ones); diagonal = ones = None
930 tmp = torch.ones(4)
936 self.assert_functionalization(f, torch.ones(4, 2), mutated_input_metadata=True)
937 logs = self.get_logs(f, torch.ones(4, 2))
945 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
948 add = torch.ops.aten.add.Tensor(slice_copy, ones); slice_copy = ones = None
961 f, torch.ones(4, 2), reapply_views=True, run_reinplace=False
970 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
973 add = torch.ops.aten.add.Tensor(slice_1, ones); slice_1 = ones = None
987 tmp = torch.ones(4)
993 self.assert_functionalization(f, torch.ones(4, 2), mutated_input_metadata=True)
994 logs = self.get_logs(f, torch.ones(4, 2))
1002 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
1005 add = torch.ops.aten.add.Tensor(select_copy, ones); select_copy = ones = None
1018 f, torch.ones(4, 2), reapply_views=True, run_reinplace=False
1027 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
1030 add = torch.ops.aten.add.Tensor(select, ones); select = ones = None
1044 tmp = torch.ones(4)
1050 self.assert_functionalization(f, torch.ones(4, 2), mutated_input_metadata=True)
1051 logs = self.get_logs(f, torch.ones(4, 2))
1059 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
1064 add = torch.ops.aten.add.Tensor(getitem, ones); getitem = ones = None
1079 f, torch.ones(4, 2), reapply_views=True, run_reinplace=False
1088 ones = torch.ops.aten.ones.default([4], device = device(type='cpu'), pin_memory = False)
1093 add = torch.ops.aten.add.Tensor(getitem, ones); getitem = ones = None
1116 self.assert_functionalization(f, torch.ones(4, 2))
1117 logs = self.get_logs(f, torch.ones(4, 2))
1139 tmp = torch.ones(4, 2)
1146 self.assert_functionalization(f, torch.ones(4, 2))
1147 logs = self.get_logs(f, torch.ones(4, 2))
1155 …ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False); ones…
1176 self.assert_functionalization(f, torch.ones(4, 2))
1177 logs = self.get_logs(f, torch.ones(4, 2))
1193 f, torch.ones(2, 2), reapply_views=True, run_reinplace=True
1212 out_1 = torch.ones(1)
1236 logs = self.get_logs(f, torch.ones(4, 2))
1252 tmp = torch.ones(2, 2)
1264 self.assert_functionalization(f, torch.ones(4, 2))
1265 logs = self.get_logs(f, torch.ones(4, 2))
1273 ones = torch.ops.aten.ones.default([2, 2], device = device(type='cpu'), pin_memory = False)
1283 add_1 = torch.ops.aten.add.Tensor(getitem, ones); getitem = ones = None
1323 f, torch.ones(4, 2), reapply_views=True, run_reinplace=True
1332 ones = torch.ops.aten.ones.default([2, 2], device = device(type='cpu'), pin_memory = False)
1342 add_1 = torch.ops.aten.add_.Tensor(getitem, ones); getitem = ones = add_1 = None
1374 tmp = torch.ones(4, 2)
1380 self.assert_functionalization(f, torch.ones(4, 2), reapply_views=True)
1381 logs = self.get_logs(f, torch.ones(4, 2), reapply_views=True)
1389 ones = torch.ops.aten.ones.default([4, 2], device = device(type='cpu'), pin_memory = False)
1391 add = torch.ops.aten.add.Tensor(view, ones); view = ones = None
1402 tmp = torch.ones(4, 2)
1408 input_functional = torch._to_functional_tensor(torch.ones(4, 2))
1434 # self.assert_functionalization(f, torch.ones(2))
1435 logs = self.get_logs(f, torch.ones(2))
1456 f, torch.ones(2), reapply_views=True, run_reinplace=True
1476 self.assert_functionalization(f, torch.ones(1))
1477 logs = self.get_logs(f, torch.ones(1))
1498 f, torch.ones(1), reapply_views=True, run_reinplace=True
1518 self.assert_functionalization(f, torch.ones(2, dtype=torch.long))
1519 logs = self.get_logs(f, torch.ones(2, dtype=torch.long))
1540 f, torch.ones(2, dtype=torch.long), reapply_views=True, run_reinplace=True
1560 self.assert_functionalization(f, torch.ones(1, dtype=torch.long))
1561 logs = self.get_logs(f, torch.ones(1, dtype=torch.long))
1582 f, torch.ones(1, dtype=torch.long), reapply_views=True, run_reinplace=True
1607 self.assert_functionalization(f, torch.ones(2, 2))
1608 logs = self.get_logs(f, torch.ones(2, 2))
1628 self.assert_functionalization(f, torch.ones(2, 2))
1629 logs = self.get_logs(f, torch.ones(2, 2))
1647 f, torch.ones(2, 2), reapply_views=True, run_reinplace=True
1675 self.assert_functionalization(f, torch.ones(8, 2))
1676 logs = self.get_logs(f, torch.ones(8, 2))
1706 f, torch.ones(8, 2), reapply_views=True, run_reinplace=True
1741 self.assert_functionalization(f, torch.ones(5, 5, 5))
1760 self.assert_functionalization(f, torch.ones(8, 2))
1761 logs = self.get_logs(f, torch.ones(8, 2))
1781 f, torch.ones(8, 2), reapply_views=True, run_reinplace=True
1819 self.assert_functionalization(f, torch.ones(8, 2))
1835 self.assert_functionalization(f, torch.ones(2, 2))
1843 x1_not_functional = LoggingTensor(torch.ones(4))
1844 x2_functional = torch._to_functional_tensor(LoggingTensor(torch.ones(4)))
1859 x1_not_functional = torch.ones(4)
1860 x2_functional = torch._to_functional_tensor(torch.ones(4))
1874 self.assert_functionalization(f, torch.ones(2))
1875 logs = self.get_logs(f, torch.ones(2))
1893 f, torch.ones(2), reapply_views=True, run_reinplace=True
1928 f, torch.randn(20, size, 35, 45), torch.zeros(size), torch.ones(size)
1935 f, torch.randn(20, size, 35, 45), torch.zeros(size), torch.ones(size)
1979 torch.ones(size),
2033 x = torch.ones(1, 5)
2044 f, torch.randn(20, 100, 35, 45), torch.zeros(100), torch.ones(100)
2047 f, torch.randn(20, 100, 35, 45), torch.zeros(100), torch.ones(100)
2073 torch.ones(100),