ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Frontend.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2019 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 __MOCO_TENSORFLOW_FRONTEND_H__
18
#define __MOCO_TENSORFLOW_FRONTEND_H__
19
20
#include <
moco/Import/ModelSignature.h
>
21
22
#include <
loco.h
>
23
24
#include <tensorflow/core/framework/graph.pb.h>
25
26
namespace
moco
27
{
28
namespace
tf
29
{
30
31
class
Frontend
32
{
33
public
:
34
enum class
FileType
35
{
36
Text
,
37
Binary
,
38
};
39
40
public
:
41
Frontend
();
42
43
public
:
44
std::unique_ptr<loco::Graph>
load
(
const
ModelSignature
&,
const
char
*,
FileType
)
const
;
45
std::unique_ptr<loco::Graph>
load
(
const
ModelSignature
&, std::istream *,
FileType
)
const
;
46
47
private
:
48
std::unique_ptr<loco::Graph>
import
(
const
ModelSignature
&, tensorflow::GraphDef &)
const
;
49
};
50
51
}
// namespace tf
52
}
// namespace moco
53
54
#endif
// __MOCO_TENSORFLOW_FRONTEND_H__
ModelSignature.h
moco::tf::Frontend
Definition
Frontend.h:32
moco::tf::Frontend::load
std::unique_ptr< loco::Graph > load(const ModelSignature &, const char *, FileType) const
Definition
Frontend.cpp:193
moco::tf::Frontend::Frontend
Frontend()
Definition
Frontend.cpp:188
moco::tf::Frontend::FileType
FileType
Definition
Frontend.h:35
moco::tf::Frontend::FileType::Binary
@ Binary
moco::tf::Frontend::FileType::Text
@ Text
loco.h
moco
Definition
Log.h:23
moco::ModelSignature
Class to store information to run a model. Normally this info comes from users via CLI params or conf...
Definition
ModelSignature.h:36
compiler
moco-tf
include
moco
tf
Frontend.h
Generated by
1.9.8