ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions
caffeimport::GraphBuilderRegistry Class Reference

#include <GraphBuilderRegistry.h>

Public Member Functions

const GraphBuilder * lookup (const std::string &layer) const
 

Static Public Member Functions

static GraphBuilderRegistry & get ()
 

Detailed Description

Definition at line 28 of file GraphBuilderRegistry.h.

Member Function Documentation

◆ get()

static GraphBuilderRegistry & caffeimport::GraphBuilderRegistry::get ( )
inlinestatic

Definition at line 39 of file GraphBuilderRegistry.h.

40 {
41 static GraphBuilderRegistry me;
42 return me;
43 }

Referenced by Frontend::load().

◆ lookup()

const GraphBuilder * caffeimport::GraphBuilderRegistry::lookup ( const std::string &  layer) const
inline

Definition at line 31 of file GraphBuilderRegistry.h.

32 {
33 if (_builder_map.find(layer) == _builder_map.end())
34 return nullptr;
35
36 return _builder_map.at(layer).get();
37 }
The documentation for this class was generated from the following files: