From 526ff04c7ece709611956f573ea97bb404ec6b06 Mon Sep 17 00:00:00 2001 From: Ziheng Jiang Date: Fri, 10 Feb 2017 09:14:11 -0800 Subject: [PATCH] [PYTHON/API] Add compare and logic build-in op for Expr (#39) * [PYTHON/API] Add compare and logic build-in op for Expr * remove 'and', 'or' --- python/tvm/expr.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/python/tvm/expr.py b/python/tvm/expr.py index c3b0845aecb6..8ae1d45d8b5e 100644 --- a/python/tvm/expr.py +++ b/python/tvm/expr.py @@ -37,6 +37,24 @@ def __rtruediv__(self, other): def __neg__(self): return self.__mul__(-1) + def __lt__(self, other): + return _make.LT(self, other) + + def __le__(self, other): + return _make.LE(self, other) + + def __eq__(self, other): + return _make.EQ(self, other) + + def __ne__(self, other): + return _make.NE(self, other) + + def __gt__(self, other): + return _make.GT(self, other) + + def __ge__(self, other): + return _make.GE(self, other) + class Expr(NodeBase, ExprOp): pass