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 440ca04..e10f132 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);
int idiv_floor(int input1, int input2);
@@ -318,6 +335,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;