ONE - On-device Neural Engine
Loading...
Searching...
No Matches
GraphBuilderRegistry.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2018 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 __GRAPH_BUILDER_REGISTRY_H__
18
#define __GRAPH_BUILDER_REGISTRY_H__
19
20
#include "
GraphBuilder.h
"
21
22
#include <map>
23
#include <string>
24
25
namespace
caffeimport
26
{
27
28
class
GraphBuilderRegistry
29
{
30
public
:
31
const
GraphBuilder
*
lookup
(
const
std::string &layer)
const
32
{
33
if
(_builder_map.find(layer) == _builder_map.end())
34
return
nullptr
;
35
36
return
_builder_map.at(layer).get();
37
}
38
39
static
GraphBuilderRegistry
&
get
()
40
{
41
static
GraphBuilderRegistry
me;
42
return
me;
43
}
44
45
private
:
46
GraphBuilderRegistry
();
47
48
private
:
49
std::map<std::string, std::unique_ptr<GraphBuilder>> _builder_map;
50
};
51
52
}
// namespace caffeimport
53
54
#endif
// __GRAPH_BUILDER_REGISTRY_H__
caffeimport::GraphBuilder
Definition
GraphBuilder.h:28
caffeimport::GraphBuilderRegistry
Definition
GraphBuilderRegistry.h:29
caffeimport::GraphBuilderRegistry::get
static GraphBuilderRegistry & get()
Definition
GraphBuilderRegistry.h:39
caffeimport::GraphBuilderRegistry::lookup
const GraphBuilder * lookup(const std::string &layer) const
Definition
GraphBuilderRegistry.h:31
caffeimport
Definition
Context.h:30
GraphBuilder.h
compiler
enco
frontend
caffe
src
GraphBuilderRegistry.h
Generated by
1.9.8