Home
last modified time | relevance | path

Searched refs:_check_module_exists (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A D_import_utils.py8 def _check_module_exists(name: str) -> bool: function
25 _check_module_exists("dill")
/aosp_15_r20/external/pytorch/torch/
H A Dhub.py359 def _check_module_exists(name): function
369 missing_deps = [pkg for pkg in dependencies if not _check_module_exists(pkg)]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py98 from torch.utils._import_utils import _check_module_exists
1339 TEST_NUMPY = _check_module_exists('numpy')
1340 TEST_FAIRSEQ = _check_module_exists('fairseq')
1341 TEST_SCIPY = _check_module_exists('scipy')
1350 TEST_NUMBA = _check_module_exists('numba')
1351 TEST_TRANSFORMERS = _check_module_exists('transformers')
1352 TEST_DILL = _check_module_exists('dill')
1354 TEST_LIBROSA = _check_module_exists('librosa') and not IS_ARM64
1356 TEST_OPT_EINSUM = _check_module_exists('opt_einsum')
1358 TEST_Z3 = _check_module_exists('z3')
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_no_runtime.py1069 not torch.hub._check_module_exists("torch_scatter"),
H A Dtest_pytorch_onnx_onnxruntime.py13785 not torch.hub._check_module_exists("torch_geometric"),