Lines Matching refs:node
41 def visit_Compare(self, node):
42 node = self.generic_visit(node)
43 if len(node.ops) > 1:
45 cmp_type = type(node.ops[0])
49 func=tf_function, args=[node.left, node.comparators[0]], keywords=[])
50 return node
52 def visit_UnaryOp(self, node):
53 node = self.generic_visit(node)
54 if isinstance(node.op, gast.Not):
56 node.op)]).body[0].value
57 node = gast.Call(func=tf_function, args=[node.operand], keywords=[])
58 return node
60 def visit_BoolOp(self, node):
62 node = self.generic_visit(node)
63 tf_function = parser.parse_str(self.op_mapping[type(node.op)]).body[0].value
64 left = node.values[0]
65 for i in range(1, len(node.values)):
67 func=tf_function, args=[left, node.values[i]], keywords=[])
71 def transform(node):
73 node = transformer.visit(node)
74 return node
