diff options
Diffstat (limited to 'reference_model/src/tensor.cc')
-rw-r--r-- | reference_model/src/tensor.cc | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/reference_model/src/tensor.cc b/reference_model/src/tensor.cc index 0678bbd..7af2069 100644 --- a/reference_model/src/tensor.cc +++ b/reference_model/src/tensor.cc @@ -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. @@ -32,11 +32,18 @@ TosaReference::Tensor::Tensor(std::string tensorName_, DType tensorDtype_, std:: consumers.clear(); isSubgraphInput = false; isSubgraphOutput = false; + isParentGraphOutput = false; } TosaReference::Tensor::~Tensor() {} +int TosaReference::Tensor::setIsParentGraphOutput() +{ + isParentGraphOutput = true; + return 0; +} + int TosaReference::Tensor::setIsSubgraphInput() { isSubgraphInput = true; |