Searched refs:DoubleSizeMaybeAddGeThreeTensor (Results 1 – 1 of 1) sorted by relevance
1193 class DoubleSizeMaybeAddGeThreeTensor(torch.Tensor): class1223 return DoubleSizeMaybeAddGeThreeTensor(inner_tensors["inner_elem"])1234 DoubleSizeMaybeAddGeThreeTensor, lambda x: x.inner_elem, args1242 return DoubleSizeMaybeAddGeThreeTensor(out_inner)1275 x = DoubleSizeMaybeAddGeThreeTensor(inp)