提交 a8faeda6 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!105 Remove memoize decorator to avoid generating pkl files.

Merge pull request !105 from lvwenyuan/remove_pkl_file
...@@ -21,6 +21,9 @@ ...@@ -21,6 +21,9 @@
https://arxiv.org/abs/1509.09308 https://arxiv.org/abs/1509.09308
https://github.com/andravin/wincnn https://github.com/andravin/wincnn
""" """
"""
2020.08.10 - Remove memoize decorator of function winograd_transform_matrices.
"""
from operator import mul from operator import mul
from functools import reduce from functools import reduce
...@@ -133,7 +136,6 @@ def _interpolation_points(degree): ...@@ -133,7 +136,6 @@ def _interpolation_points(degree):
return np.array(in_pts[degree-1], dtype=np.float64) return np.array(in_pts[degree-1], dtype=np.float64)
@memoize("topi.nn.winograd_matrices", save_at_exit=False)
def winograd_transform_matrices(tile_size, kernel_size, out_dtype): def winograd_transform_matrices(tile_size, kernel_size, out_dtype):
"""Compute the A, B, and G transform matrices for `tile_size` as a `tvm.Expr`. """Compute the A, B, and G transform matrices for `tile_size` as a `tvm.Expr`.
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册