diff options
Diffstat (limited to 'reference_model/src/graph_node.h')
-rw-r--r-- | reference_model/src/graph_node.h | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/reference_model/src/graph_node.h b/reference_model/src/graph_node.h index 874f1d8..b227d17 100644 --- a/reference_model/src/graph_node.h +++ b/reference_model/src/graph_node.h @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2022, ARM Limited. +// Copyright (c) 2020-2023, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -163,7 +163,6 @@ public: int addInputTensor(Tensor* tens); int addOutputTensor(Tensor* tens); - // Validate that the input tensors match properly // in their types, attributes, rank, etc well enough to be // processed. @@ -252,6 +251,22 @@ public: return nodeType; } + SubgraphTraverser* getParentSGT() + { + return parent_sgt; + } + + int setInMainBlock(bool isInMainBlock) + { + inMainBlock = isInMainBlock; + return 0; + } + + bool getInMainBlock() + { + return inMainBlock; + } + // Helper functions. int idiv_check(int input1, int input2, int& result); @@ -328,6 +343,8 @@ protected: // -1 means n/a int requiredRankMin; int requiredRankMax; + + bool inMainBlock; }; }; // namespace TosaReference |