Home
last modified time | relevance | path

Searched defs:_is_in_bad_fork (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/mps/
H A D__init__.py14 _is_in_bad_fork = getattr(torch._C, "_mps_is_in_bad_fork", lambda: False) variable
/aosp_15_r20/external/pytorch/torch/mtia/
H A D__init__.py42 def _is_in_bad_fork() -> bool: function
/aosp_15_r20/external/pytorch/torch/xpu/
H A D__init__.py29 _is_in_bad_fork = getattr(torch._C, "_xpu_isInBadFork", lambda: False) variable
/aosp_15_r20/external/pytorch/torch/cuda/
H A D__init__.py51 _is_in_bad_fork = getattr(torch._C, "_cuda_isInBadFork", lambda: False) variable