Imported Upstream version 1.12.0
[platform/core/ml/nnfw.git] / compiler / luci / pass / include / luci / Pass / TypeInferencePass.h
1
2 /*
3  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 #ifndef __LUCI_TYPE_INFERENCE_PASS_H__
19 #define __LUCI_TYPE_INFERENCE_PASS_H__
20
21 #include <loco.h>
22
23 #include <luci/ModulePass.h>
24
25 namespace luci
26 {
27
28 /**
29  * @brief Pass to infer type of nodes
30  */
31 class TypeInferencePass : public luci::Pass
32 {
33 public:
34   virtual const char *name(void) const { return "luci::TypeInferencePass"; }
35
36 public:
37   bool run(luci::Module *m);
38   bool run(loco::Graph *graph);
39 };
40
41 } // namespace luci
42
43 #endif //__LUCI_TYPE_INFERENCE_PASS_H__