_lazy.pyi 968 B

123456789101112131415161718192021222324252627
  1. from typing import List
  2. from torch import Tensor
  3. # defined in torch/csrc/lazy/python/init.cpp
  4. def _mark_step(device: str, devices: List[str], wait: bool): ...
  5. def _wait_device_ops(devices: List[str]): ...
  6. def _reset_metrics(): ...
  7. def _counter_names() -> List[str]: ...
  8. def _counter_value(name: str) -> int: ...
  9. def _metrics_report() -> str: ...
  10. def _get_graph_hash(tensors: List[Tensor]) -> str: ...
  11. def _sync_multi(
  12. tensors: List[Tensor],
  13. devices: List[str],
  14. wait: bool = True,
  15. sync_ltc_data: bool = True,
  16. ): ...
  17. def _get_tensor_id(tensor: Tensor) -> int: ...
  18. def _get_tensors_text(tensors: List[Tensor]) -> str: ...
  19. def _get_tensors_dot(tensors: List[Tensor]) -> str: ...
  20. def _get_tensors_backend(tensors: List[Tensor]) -> str: ...
  21. def _get_force_fallback() -> str: ...
  22. def _set_force_fallback(newval: str): ...
  23. def _clear_ir_cache(): ...
  24. def _dump_ir_cache(filename: str): ...
  25. def _set_reuse_ir(val: bool): ...
  26. def _get_default_device_type(): ...