Searched defs:inner_tensors (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 247 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): argument
297 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): argument
350 def __tensor_unflatten__(inner_tensors, meta, sizes, strides): argument
1222 def __tensor_unflatten__(inner_tensors, _, outer_size, outer_stride): argument
1477 def __tensor_unflatten__(inner_tensors, meta, _, __): argument
1560 def __tensor_unflatten__(inner_tensors, ctx, outer_size, outer_stride): argument
|
| /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): argument
|
| H A D | two_tensor.py | 47 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): argument
|
| H A D | common_subclass.py | 246 def __tensor_unflatten__(cls, inner_tensors, meta, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | semi_structured.py | 178 inner_tensors, argument
|
| /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): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | _functional_collectives.py | 598 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): argument
|