ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Importer.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_IMPORTER_H__
18
#define __MOCO_IMPORTER_H__
19
20
#include "
moco/Import/ModelSignature.h
"
21
#include "
moco/Import/GraphBuilderRegistry.h
"
22
23
#include <
moco/Names.h
>
24
25
#include <
loco.h
>
26
27
#include <tensorflow/core/framework/graph.pb.h>
28
29
#include <memory>
30
31
namespace
moco
32
{
33
34
class
Importer
final
35
{
36
public
:
37
Importer
();
38
39
public
:
40
explicit
Importer
(
const
GraphBuilderSource
*source) : _source{source}
41
{
42
// DO NOTHING
43
}
44
45
public
:
46
std::unique_ptr<loco::Graph>
import
(
const
ModelSignature
&, tensorflow::GraphDef &)
const
;
47
48
private
:
49
const
GraphBuilderSource
*_source =
nullptr
;
50
};
51
52
}
// namespace moco
53
54
#endif
// __MOCO_IMPORTER_H__
ModelSignature.h
Names.h
moco::Importer
Definition
Importer.h:35
moco::Importer::Importer
Importer()
Definition
Importer.cpp:173
moco::Importer::Importer
Importer(const GraphBuilderSource *source)
Definition
Importer.h:40
loco.h
GraphBuilderRegistry.h
moco
Definition
Log.h:23
moco::GraphBuilderSource
Definition
GraphBuilderRegistry.h:30
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
import
include
moco
Importer.h
Generated by
1.9.8