Imported Upstream version 1.25.0
[platform/core/ml/nnfw.git] / runtime / onert / backend / train / Tensor.h
1 /*
2  * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef __ONERT_BACKEND_TRAIN_TENSOR_H__
18 #define __ONERT_BACKEND_TRAIN_TENSOR_H__
19
20 #include <backend/basic/Tensor.h>
21 #include <backend/basic/train/TrainableTensor.h>
22
23 namespace onert
24 {
25 namespace backend
26 {
27 namespace train
28 {
29
30 // NOTE This class can be replaced with basic::Tensor if this backend supports dynamic tensors.
31 class Tensor : public basic::Tensor
32 {
33 public:
34   Tensor() = delete;
35
36 public:
37   Tensor(const ir::OperandInfo &info, const ir::Layout layout)
38     : basic::Tensor{info, layout, nullptr}
39   {
40     // DO NOTHING
41   }
42
43 public:
44   bool applyShape(const ir::Shape &) override { return false; }
45 };
46
47 using TrainableTensor = basic::train::TrainableTensor;
48 using DerivativeTensor = Tensor;
49 using GradientTensor = Tensor;
50
51 } // namespace train
52 } // namespace backend
53 } // namespace onert
54
55 #endif // __ONERT_BACKEND_TRAIN_TENSOR_H__