ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions
tflimport::GraphBuilder Class Referenceabstract

Parent class of tflite operation graph builders (e.g., Conv2DGraphBuilder) More...

#include <GraphBuilder.h>

Public Member Functions

virtual bool validate (const tflite::Operator *) const
 
virtual void build (const tflite::Operator *op, GraphBuilderContext *context) const =0
 
virtual ~GraphBuilder ()
 

Detailed Description

Parent class of tflite operation graph builders (e.g., Conv2DGraphBuilder)

Definition at line 30 of file GraphBuilder.h.

Constructor & Destructor Documentation

◆ ~GraphBuilder()

virtual tflimport::GraphBuilder::~GraphBuilder ( )
inlinevirtual

Definition at line 41 of file GraphBuilder.h.

41{}

Member Function Documentation

◆ build()

virtual void tflimport::GraphBuilder::build ( const tflite::Operator *  op,
GraphBuilderContext *  context 
) const
pure virtual

Implemented in tflimport::AvgPool2DGraphBuilder, tflimport::ConcatenationGraphBuilder, tflimport::DivGraphBuilder, tflimport::MaxPool2DGraphBuilder, tflimport::ReLUGraphBuilder, tflimport::ReLU6GraphBuilder, tflimport::ReshapeGraphBuilder, tflimport::SubGraphBuilder, tflimport::Conv2DGraphBuilder, and tflimport::DepthwiseConv2DGraphBuilder.

◆ validate()

virtual bool tflimport::GraphBuilder::validate ( const tflite::Operator *  ) const
inlinevirtual

TODO Declare "validate" method as a pure virtual method

Q: Is it possible to validate T/F Lite model only with this interface?

Reimplemented in tflimport::AvgPool2DGraphBuilder, tflimport::Conv2DGraphBuilder, tflimport::DepthwiseConv2DGraphBuilder, and tflimport::MaxPool2DGraphBuilder.

Definition at line 38 of file GraphBuilder.h.

38{ return true; }
The documentation for this class was generated from the following file: