11import abc
22from _typeshed import Incomplete , Unused
33from abc import ABC , ABCMeta , abstractmethod
4- from builtins import bool as _bool
4+ from builtins import bool as _bool , slice as _slice
55from collections .abc import Callable , Generator , Iterable , Iterator , Sequence
66from contextlib import contextmanager
77from enum import Enum
@@ -239,7 +239,7 @@ class TensorShape(metaclass=ABCMeta):
239239 @overload
240240 def __getitem__ (self , key : int ) -> int | None : ...
241241 @overload
242- def __getitem__ (self , key : slice ) -> TensorShape : ...
242+ def __getitem__ (self , key : _slice ) -> TensorShape : ...
243243
244244 def __iter__ (self ) -> Iterator [int | None ]: ...
245245 def __len__ (self ) -> int : ...
@@ -506,6 +506,7 @@ def clip_by_value(
506506 t : Tensor | IndexedSlices , clip_value_min : TensorCompatible , clip_value_max : TensorCompatible , name : str | None = None
507507) -> Tensor : ...
508508def tile (input : RaggedTensorLike , multiples : Tensor | Sequence [int ], name : str | None = None ) -> Tensor : ...
509+
509510@overload
510511def range (
511512 limit : int | Tensor , / , * , delta : int | Tensor = 1 , dtype : DTypeLike | None = None , name : str | None = "range"
@@ -518,4 +519,5 @@ def range(
518519 dtype : DTypeLike | None = None ,
519520 name : str | None = "range" ,
520521) -> Tensor : ...
522+
521523def __getattr__ (name : str ): ... # incomplete module
0 commit comments