add skip ops: shape/reshape2

241d4c8a · chenluyan · b6659ca8 · 241d4c8a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddleslim/prune/prune_walker.py paddleslim/prune/prune_walker.py +1 -1

未找到文件。
--- a/paddleslim/prune/prune_walker.py
+++ b/paddleslim/prune/prune_walker.py
@@ -23,7 +23,7 @@ _logger = get_logger(__name__, level=logging.INFO)

 PRUNE_WORKER = Registry('prune_worker')

-SKIP_OPS = ["conditional_block"]
+SKIP_OPS = ["conditional_block","shape","reshape2"]


 class PruneWorker(object):