_lazy_ts_backend.pyi 303 B

12345678
  1. #defined in torch/csrc/lazy/python/init.cpp
  2. from typing import List, Tuple, Any
  3. from torch import Tensor
  4. def _init(): ...
  5. def _get_tensors_ts_device_data_node(tensors: List[Tensor]) -> Tuple[List[int], List[Any]]: ...
  6. def _run_cached_graph(hash_str: str, graph_inputs: List[Any]) -> List[Tensor]: ...