Searched refs:_unwrap_functional_tensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/functorch/_src/eager_transforms/ |
| H A D | __init__.py | 6 _unwrap_functional_tensor,
|
| /aosp_15_r20/external/pytorch/torch/csrc/functorch/ |
| H A D | init.cpp | 170 Tensor _unwrap_functional_tensor(const Tensor& self, bool add_back_views) { in _unwrap_functional_tensor() function
466 &_unwrap_functional_tensor, in initFuncTorchBindings()
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | _functorch.pyi | 27 def _unwrap_functional_tensor(tensor: Tensor, reapply_views: bool) -> Tensor: ...
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | eager_transforms.py | 25 _unwrap_functional_tensor,
1497 return _unwrap_functional_tensor(maybe_tensor, reapply_views)
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 28 _unwrap_functional_tensor,
284 _unwrap_functional_tensor(t, reapply_views), trace=trace
|
| H A D | fake_tensor.py | 184 unwrapped = torch._C._functorch._unwrap_functional_tensor(x, reapply_views)
205 unwrapped = torch._C._functorch._unwrap_functional_tensor(t, reapply_views)
|