Lines Matching refs:LoggingTensor

37     LoggingTensor,
766             x = LoggingTensor(torch.tensor([3.0]), requires_grad=True)
770             grad_y = LoggingTensor(torch.tensor([1.0]))
795             x = LoggingTensor(torch.ones(1))
796             y = LoggingTensor(torch.zeros(1))
814             x = LoggingTensor(torch.ones(1))
815             y = LoggingTensor(torch.ones(1, 1))
816             z = LoggingTensor(torch.ones(1))
843             x = LoggingTensor(torch.ones(1))
864             x = LoggingTensor(torch.ones(2, 2))
894                 x = LoggingTensor(torch.ones(2, 2))
952             x = LoggingTensor(torch.tensor([3.0]), requires_grad=True)
970         x = LoggingTensor(torch.ones(1))
977         torch.Tensor._make_wrapper_subclass(LoggingTensor, (1000000000000,))
980         x = LoggingTensor(torch.ones(1))
1030         x = LoggingTensor(torch.ones(1))
1050                 assert isinstance(grad_output, LoggingTensor)
1052                 assert isinstance(x, LoggingTensor)
1057             x = LoggingTensor(torch.ones(1), requires_grad=True)
1059             x.grad = LoggingTensor(torch.zeros(1))
1062             grad_output = LoggingTensor(torch.ones(1))
1097             a = torch.Tensor._make_subclass(Foo, LoggingTensor(torch.rand(2)))
1099             b = LoggingTensor(torch.rand(2)).as_subclass(Foo)
1101             Foo(LoggingTensor(torch.rand(2)))
1204             x = LoggingTensor(torch.randperm(3))
1207             with torch.serialization.safe_globals([LoggingTensor]):
1216         x = LoggingTensor(torch.randperm(3))
1784         x = LoggingTensor(torch.tensor([2.0, 3.0]))
1934         z = LoggingTensor(torch.empty([]))