| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 247 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): member in ScaledTensor
297 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): member in OptionalScaledTensor
350 def __tensor_unflatten__(inner_tensors, meta, sizes, strides): member in CtxSubclassTensor
1222 def __tensor_unflatten__(inner_tensors, _, outer_size, outer_stride): member in SubclassTests.test_wrapper_subclass_guards_on_inner_tensor.DoubleSizeMaybeAddGeThreeTensor
1477 def __tensor_unflatten__(inner_tensors, meta, _, __): member in SubclassTests.test_subclass_constructor_proxying.SubclassTensor
1560 def __tensor_unflatten__(inner_tensors, ctx, outer_size, outer_stride): member in SubclassTests.test_torch_function_subclass_survives_into_aot_autograd.SubTensor
|
| H A D | test_autograd_function.py | 894 def __tensor_unflatten__(tensors, metadatas, outer_size, outer_stride): member in AutogradFunctionTests.test_tensor_subclass_intermediary_input.FooTensor
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | custom_tensor.py | 37 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): member in ConstantExtraMetadataTensor
|
| H A D | two_tensor.py | 47 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): member in TwoTensor
|
| H A D | common_subclass.py | 246 def __tensor_unflatten__(cls, inner_tensors, meta, outer_size, outer_stride): member in SubclassWithTensorFactory
|
| /aosp_15_r20/external/executorch/examples/models/llama/experimental/ |
| H A D | subclass.py | 196 def __tensor_unflatten__( member in GGMLInt4LinearWeight
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | semi_structured.py | 176 def __tensor_unflatten__( member in SparseSemiStructuredTensor
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | nested_tensor.py | 250 def __tensor_unflatten__(inner_tensors: Dict, meta, outer_size, outer_stride): member in NestedTensor
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 300 …def __tensor_unflatten__(inner_tensors: int, flatten_spec: int, outer_size: int, outer_stride: int… member in TensorWithFlatten
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _api.py | 298 def __tensor_unflatten__(inner_tensors, flatten_spec, outer_size, outer_stride): member in DTensor
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | _functional_collectives.py | 598 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): member in AsyncCollectiveTensor
|