aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
Diffstat (limited to 'include')
-rw-r--r--include/operator.def1
-rw-r--r--include/schema_operator.def1
2 files changed, 2 insertions, 0 deletions
diff --git a/include/operator.def b/include/operator.def
index 10cbc5d..6198c0e 100644
--- a/include/operator.def
+++ b/include/operator.def
@@ -91,6 +91,7 @@ DEF_OPERATOR(ReduceSum)
/* memory operation */
DEF_OPERATOR(Concat)
DEF_OPERATOR(Pad)
+DEF_OPERATOR(Dim)
DEF_OPERATOR(Reshape)
DEF_OPERATOR(Reverse)
DEF_OPERATOR(Slice)
diff --git a/include/schema_operator.def b/include/schema_operator.def
index a11eeeb..52c7ae4 100644
--- a/include/schema_operator.def
+++ b/include/schema_operator.def
@@ -75,6 +75,7 @@ DEF_SCHEMA_OPERATOR(REDUCE_PRODUCT)
DEF_SCHEMA_OPERATOR(REDUCE_SUM)
DEF_SCHEMA_OPERATOR(CONCAT)
DEF_SCHEMA_OPERATOR(PAD)
+DEF_SCHEMA_OPERATOR(DIM)
DEF_SCHEMA_OPERATOR(RESHAPE)
DEF_SCHEMA_OPERATOR(REVERSE)
DEF_SCHEMA_OPERATOR(SLICE)