Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py1193 class DoubleSizeMaybeAddGeThreeTensor(torch.Tensor): class
1223 return DoubleSizeMaybeAddGeThreeTensor(inner_tensors["inner_elem"])
1234 DoubleSizeMaybeAddGeThreeTensor, lambda x: x.inner_elem, args
1242 return DoubleSizeMaybeAddGeThreeTensor(out_inner)
1275 x = DoubleSizeMaybeAddGeThreeTensor(inp)