pytorch/torch/_dynamo/polyfills/builtins.py
2024-08-26 15:16:17 +00:00

31 lines
509 B
Python

"""
Python polyfills for builtins
"""
import builtins
from typing import Iterable
from ..decorators import substitute_in_graph
__all__ = [
"all",
"any",
]
@substitute_in_graph(builtins.all)
def all(iterable: Iterable[object], /) -> bool:
for elem in iterable:
if not elem:
return False
return True
@substitute_in_graph(builtins.any)
def any(iterable: Iterable[object], /) -> bool:
for elem in iterable:
if elem:
return True
return False