aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/subgraph_traverser.h
diff options
context:
space:
mode:
authorKevin Cheng <kevin.cheng@arm.com>2021-10-14 17:09:57 -0700
committerKevin Cheng <kevin.cheng@arm.com>2021-10-18 18:50:08 +0000
commitcc61be36c3b0f5cd1ea719e129a54fd48a6ee9a2 (patch)
tree2d664f87e3fdd75de8c6794f6f6c8d6364ece6bb /reference_model/src/subgraph_traverser.h
parente807aae606a78d923a2565052f7c2179e3050650 (diff)
downloadreference_model-cc61be36c3b0f5cd1ea719e129a54fd48a6ee9a2.tar.gz
More ERROR_IF supports
- Also delay tensor allocation after operator being validated ERROR_IF can be caught first before 0 or negative dimension set the graph_status to UNPREDICTABLE - Rescale, Argmax, FullyConnected, Matmul, Pad, Reshape, Slice, Transpose, Clamp, Concat, Equal, Greater, GreaterEqual, Table Signed-off-by: Kevin Cheng <kevin.cheng@arm.com> Change-Id: I4e1b3e5794fe195ce1a37e28443ae584645a3b91
Diffstat (limited to 'reference_model/src/subgraph_traverser.h')
-rw-r--r--reference_model/src/subgraph_traverser.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h
index 4be6c1f..d53a4c0 100644
--- a/reference_model/src/subgraph_traverser.h
+++ b/reference_model/src/subgraph_traverser.h
@@ -21,6 +21,7 @@
#include "ops/op_factory.h"
#include "tensor.h"
#include "tosa_serialization_handler.h"
+#include <unordered_set>
namespace TosaReference
{
@@ -54,6 +55,7 @@ public:
int linkTensorsAndNodes();
int validateGraph();
+ int allocateTensor();
int dumpGraph(FILE* out) const;
int dumpNextNodeList(FILE* out) const;
@@ -99,6 +101,9 @@ private:
// lifetime, although the list itself should only contain unique nodes.
std::list<GraphNode*> nextNodeList;
+ // tensor name set which contains all the name used by operator
+ std::unordered_set<std::string> used_tensor_name_set;
+
// Maximum number of times to evalute a node before
// warning.
const int MAX_EVAL_COUNT = 10000;