From 241d4c8a6a8049e2ffa9d0d8fad4b5aa25a209a5 Mon Sep 17 00:00:00 2001
From: chenluyan <chenluyan@baidu.com>
Date: Tue, 18 Aug 2020 20:35:47 +0800
Subject: [PATCH] add skip ops: shape/reshape2

---
 paddleslim/prune/prune_walker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddleslim/prune/prune_walker.py b/paddleslim/prune/prune_walker.py
index c40c617e..189b75a7 100644
--- 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):
-- 
GitLab