Skip to content

Commit

Permalink
[onert] Introduce ExtraTensor (#13576)
Browse files Browse the repository at this point in the history
This PR introduces ExtraTensor.
ExtraTensor means that the tensor is only used by one operation.

ONE-DCO-1.0-Signed-off-by: seunghui youn <[email protected]>
  • Loading branch information
zetwhite authored Aug 2, 2024
1 parent b66ccf0 commit 5457886
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions runtime/onert/core/include/backend/train/ExtraTensor.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
/*
* Copyright (c) 2024 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#ifndef __ONERT_BACKEND_EXTRA_H__
#define __ONERT_BACKEND_EXTRA_H__

#include <backend/basic/Tensor.h>

namespace onert
{
namespace backend
{
namespace train
{

// ExtraTensor is a tensor that is accessed within one operation layer.
// In other words, the scope of the extra tensor is confined to one specific layer.
class ExtraTensor final : public basic::Tensor
{
public:
ExtraTensor() = delete;

public:
ExtraTensor(const ir::OperandInfo &info) : basic::Tensor(info, nullptr)
{
// DO NOTHING
}
};

} // namespace train
} // namespace backend
} // namespace onert

#endif // __ONERT_BACKEND_EXTRA_TENSOR_H__

0 comments on commit 5457886

Please sign in to comment.