未验证 提交 7d416161 编写于 作者: Y Yiqun Liu 提交者: GitHub

Fix the count of number of float32 operators. (#52029)

上级 12cd05b4
......@@ -63,11 +63,11 @@ class KernelRegistryStatistics:
}
def update(self, supported_dtypes):
for dtype in self.num_ops_for_dtypes.keys():
if dtype in ["float", "float32"]:
self.num_ops_for_dtypes["float32"] += 1
elif dtype in supported_dtypes:
for dtype in supported_dtypes:
if dtype in self.num_ops_for_dtypes.keys():
self.num_ops_for_dtypes[dtype] += 1
elif dtype == "float":
self.num_ops_for_dtypes["float32"] += 1
self.num_ops_for_dtypes["all"] += 1
def __str__(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册