aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/numeric_util.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/numeric_util.py')
-rw-r--r--ethosu/vela/numeric_util.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/ethosu/vela/numeric_util.py b/ethosu/vela/numeric_util.py
index 4ebef8e5..20aa4a05 100644
--- a/ethosu/vela/numeric_util.py
+++ b/ethosu/vela/numeric_util.py
@@ -93,3 +93,13 @@ def full_shape(dim, shape, fill):
def overlaps(start1, end1, start2, end2):
return start1 < end2 and start2 < end1
+
+
+def is_integer(num):
+ if isinstance(num, (int, np.integer)):
+ return True
+ if type(num) is float and num.is_integer():
+ return True
+ if isinstance(num, np.inexact) and np.mod(num, 1) == 0:
+ return True
+ return False