diff options
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
-rw-r--r-- | reference_model/src/subgraph_traverser.h | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h index 3f4eecf..4be6c1f 100644 --- a/reference_model/src/subgraph_traverser.h +++ b/reference_model/src/subgraph_traverser.h @@ -16,15 +16,22 @@ #ifndef SUBGRAPH_TRAVERSER_H #define SUBGRAPH_TRAVERSER_H -#include "model_common.h" - #include "graph_node.h" +#include "model_common.h" #include "ops/op_factory.h" +#include "tensor.h" #include "tosa_serialization_handler.h" namespace TosaReference { +enum class GraphStatus : int +{ + TOSA_VALID = 0, + TOSA_UNPREDICTABLE = 1, + TOSA_ERROR = 2, +}; + class SubgraphTraverser { public: @@ -36,6 +43,15 @@ public: int evaluateNextNode(); int evaluateAll(); + GraphStatus getGraphStatus() const + { + return graph_status; + } + void setGraphStatus(GraphStatus status) + { + graph_status = status; + } + int linkTensorsAndNodes(); int validateGraph(); @@ -59,6 +75,8 @@ private: GraphNode* getNextNode(); + GraphStatus graph_status; + // pointer to serialization library and corresponding basic block TosaSerializationBasicBlock* block; TosaSerializationHandler* tsh; |