Searched refs:CtxSubclassTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 320 class CtxSubclassTensor(torch.Tensor): class
353 return CtxSubclassTensor(a, constant)
365 if isinstance(x, CtxSubclassTensor)
369 lambda x: x.a if isinstance(x, CtxSubclassTensor) else x, args
372 lambda x: x.a if isinstance(x, CtxSubclassTensor) else x, kwargs
376 lambda x: CtxSubclassTensor(x, biggest_constant)
1374 x = CtxSubclassTensor(torch.ones(2), 3)
1375 x2 = CtxSubclassTensor(torch.ones(2), 3)
1376 x3 = CtxSubclassTensor(torch.ones(2), 4)
1382 x1 = CtxSubclassTensor(x0.clone(), 2)
[all …]
|