Home
last modified time | relevance | path

Searched refs:LoggingTensor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py37 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))
[all …]
H A Dtest_subclass.py27 from torch.testing._internal.logging_tensor import LoggingTensor
177 if tensor_cls in [LoggingTensor, DiagTensorBelow]:
262 from torch.testing._internal.logging_tensor import LoggingTensor
264 x_log = LoggingTensor(x)
H A Dtest_functionalization.py27 from torch.testing._internal.logging_tensor import capture_logs, LoggingTensor
155 LoggingTensor(torch.randn(2, 2))
1843 x1_not_functional = LoggingTensor(torch.ones(4))
1844 x2_functional = torch._to_functional_tensor(LoggingTensor(torch.ones(4)))
H A Dtest_prims.py24 from torch.testing._internal.logging_tensor import LoggingTensor, capture_logs, log_input
329 r = LoggingTensor(r)
H A Dtest_proxy_tensor.py311 from torch.testing._internal.logging_tensor import LoggingTensor
319 x = LoggingTensor(x)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dlogging_tensor.py47 class LoggingTensor(torch.Tensor): class
94 class LoggingTensorReentrant(LoggingTensor):
117 cond_cls = torch.Tensor if self.use_shortid_for_all_tensors else LoggingTensor
H A Dcommon_subclass.py10 from torch.testing._internal.logging_tensor import LoggingTensor
207 LoggingTensor: SubclassInfo(
209 create_fn=lambda shape: LoggingTensor(torch.randn(shape))
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_compiled_autograd.py543 from torch.testing._internal.logging_tensor import LoggingTensor
558 LoggingTensor(torch.ones(1)),
1104 from torch.testing._internal.logging_tensor import LoggingTensor
1119 LoggingTensor(torch.ones(1)),
/aosp_15_r20/external/pytorch/test/autograd/
H A Dtest_functional.py19 from torch.testing._internal.logging_tensor import LoggingTensor
41 return LoggingTensor(ctor(*args, **kwargs), requires_grad=requires_grad)
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dextending.rst733 class LoggingTensor(torch.Tensor):