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

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

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