pytorch

1
# mypy: allow-untyped-defs
2
# defined in torch/csrc/lazy/python/init.cpp
3

4
from typing import Any
5

6
from torch import Tensor
7

8
def _init(): ...
9
def _get_tensors_ts_device_data_node(
10
    tensors: list[Tensor],
11
) -> tuple[list[int], list[Any]]: ...
12
def _run_cached_graph(hash_str: str, graph_inputs: list[Any]) -> list[Tensor]: ...
13