1 files changed, 8 insertions, 1 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h
index 08e865a..d5f1de8 100644
--- a/reference_model/src/tensor.h
+++ b/reference_model/src/tensor.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.
@@ -40,6 +40,11 @@ public:
 
     int setIsSubgraphInput();
     int setIsSubgraphOutput();
+    int setIsParentGraphOutput();
+
+    int getIsParentGraphOutput() const {
+        return isParentGraphOutput;
+    }
 
     int getIsSubgraphInput() const
     {
@@ -261,6 +266,8 @@ protected:
     int isSubgraphOutput;
     bool isAllocated;
 
+    bool isParentGraphOutput;
+
     GraphNode* producer;
     std::vector<GraphNode*> consumers;