Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py1390 class CtxSubclassTensorCustomGuardFn(CtxSubclassTensor): class
1395 x = CtxSubclassTensorCustomGuardFn(torch.ones(2), 2)
1396 x2 = CtxSubclassTensorCustomGuardFn(torch.ones(2), 3)
1397 x3 = CtxSubclassTensorCustomGuardFn(torch.ones(2), 1)
1404 class CtxSubclassTensorCustomGuardFn(CtxSubclassTensor): class
1410 x = CtxSubclassTensorCustomGuardFn(torch.ones(2), 3)
1420 class CtxSubclassTensorCustomGuardFn(CtxSubclassTensor): class
1424 x = CtxSubclassTensorCustomGuardFn(torch.ones(2), 3)