提交 d3bc076f authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Replace defaultdict(int) with Counter in rewriting/elemwise

上级 210984dd
import itertools import itertools
import sys import sys
from collections import defaultdict, deque from collections import Counter, defaultdict, deque
from collections.abc import Generator from collections.abc import Generator
from functools import cache from functools import cache
from typing import TypeVar from typing import TypeVar
...@@ -127,7 +127,7 @@ class InplaceElemwiseOptimizer(GraphRewriter): ...@@ -127,7 +127,7 @@ class InplaceElemwiseOptimizer(GraphRewriter):
"nb_call_replace": 0, "nb_call_replace": 0,
"nb_call_validate": 0, "nb_call_validate": 0,
"nb_inconsistent": 0, "nb_inconsistent": 0,
"ndim": defaultdict(int), "ndim": Counter(),
} }
check_each_change = config.tensor__insert_inplace_optimizer_validate_nb check_each_change = config.tensor__insert_inplace_optimizer_validate_nb
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论