aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/graph_node.h
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/graph_node.h')
-rw-r--r--reference_model/src/graph_node.h20
1 files changed, 20 insertions, 0 deletions
diff --git a/reference_model/src/graph_node.h b/reference_model/src/graph_node.h
index 900f4b8..b3fe8d6 100644
--- a/reference_model/src/graph_node.h
+++ b/reference_model/src/graph_node.h
@@ -254,6 +254,23 @@ public:
return inMainBlock;
}
+ int getEvaluated() const
+ {
+ return evaluated;
+ }
+
+ int setEvaluated()
+ {
+ evaluated = true;
+ return 0;
+ }
+
+ int clearEvaluated()
+ {
+ evaluated = false;
+ return 0;
+ }
+
// Helper functions.
int idiv_check(int input1, int input2, int& result);
@@ -317,6 +334,9 @@ protected:
// next-node list.
int onNextNodeList;
+ // Flag indicating that this node has been evaluated before
+ int evaluated;
+
// Required input/output tensor counts for node validation
// -1 means any number is allowed
int requiredInputCount;