ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Public Member Functions | Static Public Member Functions
tflimport::GraphBuilderRegistry Class Reference

Class to return graph builder for passed tflite::builtinOperator. More...

#include <GraphBuilderRegistry.h>

Public Member Functions

const GraphBuilder * lookup (tflite::BuiltinOperator op) const
 Returns registered GraphBuilder pointer for BuiltinOperator or nullptr if not registered.
 

Static Public Member Functions

static GraphBuilderRegistry & get ()
 

Detailed Description

Class to return graph builder for passed tflite::builtinOperator.

Definition at line 44 of file GraphBuilderRegistry.h.

Member Function Documentation

◆ get()

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

Definition at line 59 of file GraphBuilderRegistry.h.

60 {
61 static GraphBuilderRegistry me;
62 return me;
63 }

◆ lookup()

const GraphBuilder * tflimport::GraphBuilderRegistry::lookup ( tflite::BuiltinOperator  op) const
inline

Returns registered GraphBuilder pointer for BuiltinOperator or nullptr if not registered.

Definition at line 51 of file GraphBuilderRegistry.h.

52 {
53 if (_builder_map.find(op) == _builder_map.end())
54 return nullptr;
55
56 return _builder_map.at(op).get();
57 }
The documentation for this class was generated from the following file: