ONE - On-device Neural Engine
Loading...
Searching...
No Matches
GraphBuilder.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_IMPORT_GRAPH_BUILDER_H__
18
#define __MOCO_IMPORT_GRAPH_BUILDER_H__
19
20
#include "
GraphBuilderContext.h
"
21
22
#include <tensorflow/core/framework/graph.pb.h>
23
24
namespace
moco
25
{
26
30
class
GraphBuilder
31
{
32
public
:
33
virtual
bool
validate
(
const
tensorflow::NodeDef &)
const
= 0;
34
virtual
void
build
(
const
tensorflow::NodeDef &,
GraphBuilderContext
*)
const
= 0;
35
virtual
~GraphBuilder
() {}
36
};
37
38
}
// namespace moco
39
40
#endif
// __MOCO_IMPORT_GRAPH_BUILDER_H__
moco::GraphBuilderContext
Class to store context to build loco graph IR from TensorFlow.
Definition
GraphBuilderContext.h:117
moco::GraphBuilder
Interface of convert TF NodeDef to loco::Node (e.g., Conv2DGraphBuilder)
Definition
GraphBuilder.h:31
moco::GraphBuilder::~GraphBuilder
virtual ~GraphBuilder()
Definition
GraphBuilder.h:35
moco::GraphBuilder::validate
virtual bool validate(const tensorflow::NodeDef &) const =0
moco::GraphBuilder::build
virtual void build(const tensorflow::NodeDef &, GraphBuilderContext *) const =0
moco
Definition
Log.h:23
GraphBuilderContext.h
compiler
moco
import
include
moco
Import
GraphBuilder.h
Generated by
1.9.8