mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
e5621b4d8b
Pull Request resolved: https://github.com/pytorch/pytorch/pull/159368 Approved by: https://github.com/mlazos ghstack dependencies: #159365, #159366
34 lines
662 B
Python
34 lines
662 B
Python
"""
|
|
Python polyfills for builtins
|
|
"""
|
|
|
|
from collections.abc import Iterable, MutableMapping
|
|
from typing import TypeVar
|
|
|
|
from ..decorators import substitute_in_graph
|
|
|
|
|
|
__all__ = []
|
|
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
try:
|
|
import _collections # type: ignore[import-not-found]
|
|
|
|
@substitute_in_graph(_collections._count_elements)
|
|
def _count_elements(
|
|
mapping: MutableMapping[T, int],
|
|
iterable: Iterable[T],
|
|
) -> None:
|
|
"Tally elements from the iterable."
|
|
mapping_get = mapping.get
|
|
for elem in iterable:
|
|
mapping[elem] = mapping_get(elem, 0) + 1
|
|
|
|
__all__.append("_count_elements")
|
|
|
|
except ImportError:
|
|
pass
|