diff options
Diffstat (limited to 'reference_model/src/tensor.h')
-rw-r--r-- | reference_model/src/tensor.h | 9 |
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; |