ONE - On-device Neural Engine
|
Class to return graph builder for Circle nodes. More...
#include <GraphBuilderRegistry.h>
Public Member Functions | |
GraphBuilderRegistry () | |
GraphBuilderRegistry (const GraphBuilderSource *parent) | |
const GraphBuilderBase * | lookup (const circle::BuiltinOperator &op) const final |
Returns registered GraphBuilder pointer for operator or nullptr if not registered. | |
const NodeBuilderBase * | lookup (const NodeBuilderType type) const final |
Returns registered NodeBuilderBase pointer for type or nullptr if not registered. | |
void | add (const circle::BuiltinOperator op, std::unique_ptr< GraphBuilderBase > &&builder) |
void | add (std::unique_ptr< NodeBuilderBase > &&builder) |
Public Member Functions inherited from luci::GraphBuilderSource | |
virtual | ~GraphBuilderSource ()=default |
Static Public Member Functions | |
static GraphBuilderRegistry & | get () |
Class to return graph builder for Circle nodes.
Definition at line 46 of file GraphBuilderRegistry.h.
luci::GraphBuilderRegistry::GraphBuilderRegistry | ( | ) |
Definition at line 26 of file GraphBuilderRegistry.cpp.
References ADD, AVERAGE_POOL_2D, luci::CIRCLE_NODE, CONCATENATION, CONV_2D, DEPTH_TO_SPACE, DEPTHWISE_CONV_2D, DEQUANTIZE, DIV, FLOOR, FULLY_CONNECTED, L2_NORMALIZATION, L2_POOL_2D, LOCAL_RESPONSE_NORMALIZATION, LOG, LOGISTIC, MAX_POOL_2D, MUL, PAD, QUANTIZE, luci::RELU, luci::RELU6, luci::RELU_N1_TO_1, RESHAPE, RESIZE_BILINEAR, SOFTMAX, SPACE_TO_DEPTH, STRIDED_SLICE, SUB, SVDF, and luci::TANH.
|
inline |
Definition at line 52 of file GraphBuilderRegistry.h.
|
inline |
Definition at line 88 of file GraphBuilderRegistry.h.
|
inline |
Definition at line 93 of file GraphBuilderRegistry.h.
|
inlinestatic |
Definition at line 81 of file GraphBuilderRegistry.h.
Referenced by entry().
|
inlinefinalvirtual |
Returns registered GraphBuilder pointer for operator or nullptr if not registered.
Implements luci::GraphBuilderSource.
Definition at line 62 of file GraphBuilderRegistry.h.
References luci::GraphBuilderSource::lookup().
|
inlinefinalvirtual |
Returns registered NodeBuilderBase pointer for type or nullptr if not registered.
Implements luci::GraphBuilderSource.
Definition at line 73 of file GraphBuilderRegistry.h.
References luci::GraphBuilderSource::lookup().