_cpu.pyi 196 B

1234567
  1. from torch.types import _bool
  2. # Defined in torch/csrc/cpu/Module.cpp
  3. def _is_cpu_support_avx2() -> _bool: ...
  4. def _is_cpu_support_avx512() -> _bool: ...
  5. def _is_cpu_support_vnni() -> _bool: ...