Searched refs:ConstantExtraMetadataTensor (Results 1 – 2 of 2) sorted by relevance
9 class ConstantExtraMetadataTensor(torch.Tensor): class40 out = ConstantExtraMetadataTensor(elem)49 ConstantExtraMetadataTensor, lambda x: x.elem, args53 ConstantExtraMetadataTensor, lambda x: x.elem, kwargs61 ConstantExtraMetadataTensor(o_inner)
75 from torch.testing._internal.custom_tensor import ConstantExtraMetadataTensor896 custom_a = ConstantExtraMetadataTensor(a)898 custom_aa = ConstantExtraMetadataTensor(custom_a)912 self.assertTrue(isinstance(custom_aa_compile.grad, ConstantExtraMetadataTensor))914 isinstance(custom_aa_compile.grad.elem, ConstantExtraMetadataTensor)986 custom_a = ConstantExtraMetadataTensor(a)988 custom_aa = ConstantExtraMetadataTensor(custom_a)