From 1f032c53d527c12ce54ff3f8861772f9f957a26b Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Tue, 2 Jun 2020 17:15:27 +0800 Subject: [PATCH] remove duplicate function, test=develop (#24849) --- .../dygraph/dygraph_to_static/print_transformer.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/python/paddle/fluid/dygraph/dygraph_to_static/print_transformer.py b/python/paddle/fluid/dygraph/dygraph_to_static/print_transformer.py index 5c45cc8a600..e55018d2e7d 100644 --- a/python/paddle/fluid/dygraph/dygraph_to_static/print_transformer.py +++ b/python/paddle/fluid/dygraph/dygraph_to_static/print_transformer.py @@ -117,7 +117,7 @@ class PrintTransformer(gast.NodeTransformer): def _need_transform(self, var_node, print_node): if isinstance(var_node, gast.Name): - if self._is_tensor_node(var_node): + if self.static_analysis_visitor.is_tensor_node(var_node): return True else: _logger.warning( @@ -128,11 +128,3 @@ class PrintTransformer(gast.NodeTransformer): "ProgramTranslator could not transform < %s > now and will run it as-is." % ast_to_source_code(print_node).strip()) return False - - def _is_tensor_node(self, node): - tensor_types = {NodeVarType.TENSOR, NodeVarType.PADDLE_RETURN_TYPES} - wrapper_node = self.node_to_wrapper_map.get(node, None) - if wrapper_node is not None: - if wrapper_node.node_var_type & tensor_types: - return True - return False -- GitLab