Searched defs:__tensor_flatten__ (Results 1 – 11 of 11) sorted by relevance
242 def __tensor_flatten__(self): member in ScaledTensor289 def __tensor_flatten__(self): member in OptionalScaledTensor346 def __tensor_flatten__(self): member in CtxSubclassTensor1218 def __tensor_flatten__(self): member in SubclassTests.test_wrapper_subclass_guards_on_inner_tensor.DoubleSizeMaybeAddGeThreeTensor1473 def __tensor_flatten__(self): member in SubclassTests.test_subclass_constructor_proxying.SubclassTensor1556 def __tensor_flatten__(self): member in SubclassTests.test_torch_function_subclass_survives_into_aot_autograd.SubTensor
887 def __tensor_flatten__(self): member in AutogradFunctionTests.test_tensor_subclass_intermediary_input.FooTensor
30 def __tensor_flatten__(self): member in ConstantExtraMetadataTensor
43 def __tensor_flatten__(self): member in TwoTensor
242 def __tensor_flatten__(self): member in SubclassWithTensorFactory
189 def __tensor_flatten__(self): member in GGMLInt4LinearWeight
161 def __tensor_flatten__( member in SparseSemiStructuredTensor
235 def __tensor_flatten__(self): member in NestedTensor
296 def __tensor_flatten__(self) -> Tuple[Sequence[str], object]: member in TensorWithFlatten
290 def __tensor_flatten__(self): member in DTensor
591 def __tensor_flatten__(self): member in AsyncCollectiveTensor