diff --git a/third_party/incubator-tvm/topi/python/topi/nn/winograd_util.py b/third_party/incubator-tvm/topi/python/topi/nn/winograd_util.py index 464b63301b409c1e92b2e7de721475768306b06b..9bf6c3714969a4aa6173600e69ddd542b52ad850 100644 --- a/third_party/incubator-tvm/topi/python/topi/nn/winograd_util.py +++ b/third_party/incubator-tvm/topi/python/topi/nn/winograd_util.py @@ -21,6 +21,9 @@ https://arxiv.org/abs/1509.09308 https://github.com/andravin/wincnn """ +""" +2020.08.10 - Remove memoize decorator of function winograd_transform_matrices. +""" from operator import mul from functools import reduce @@ -133,7 +136,6 @@ def _interpolation_points(degree): 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): """Compute the A, B, and G transform matrices for `tile_size` as a `tvm.Expr`. """