ONE - On-device Neural Engine
Loading...
Searching...
No Matches
tflimport::GraphBuilderContext Class Reference

Class to store context to build IR from tflite. More...

#include <Context.h>

Public Member Functions

 GraphBuilderContext (coco::Module *m, coco::Data *d, coco::Block *block, TensorBags &tensor_bags, TensorContext &tensor_context, TflBufferContext &buffer_context, const tflite::SubGraph *graph)
 
 GraphBuilderContext ()=delete
 
 GraphBuilderContext (const GraphBuilderContext &)=delete
 
 GraphBuilderContext (GraphBuilderContext &&)=delete
 
coco::Modulem ()
 
coco::Datad ()
 
coco::Blockblock ()
 
TensorContexttensor ()
 
TensorBagsbags ()
 
TflBufferContextbuffer ()
 
const tflite::SubGraph * graph ()
 

Detailed Description

Class to store context to build IR from tflite.

Definition at line 132 of file Context.h.

Constructor & Destructor Documentation

◆ GraphBuilderContext() [1/4]

tflimport::GraphBuilderContext::GraphBuilderContext ( coco::Module m,
coco::Data d,
coco::Block block,
TensorBags tensor_bags,
TensorContext tensor_context,
TflBufferContext buffer_context,
const tflite::SubGraph *  graph 
)
inlineexplicit

Definition at line 135 of file Context.h.

138 : _m(m), _d(d), _block(block), _tensor_bags(tensor_bags), _tensor_context(tensor_context),
139 _buffer_context(buffer_context), _graph(graph)
140 {
141 // DO NOTHING
142 }
const tflite::SubGraph * graph()
Definition Context.h:155

◆ GraphBuilderContext() [2/4]

tflimport::GraphBuilderContext::GraphBuilderContext ( )
delete

◆ GraphBuilderContext() [3/4]

tflimport::GraphBuilderContext::GraphBuilderContext ( const GraphBuilderContext )
delete

◆ GraphBuilderContext() [4/4]

tflimport::GraphBuilderContext::GraphBuilderContext ( GraphBuilderContext &&  )
delete

Member Function Documentation

◆ bags()

◆ block()

◆ buffer()

TflBufferContext & tflimport::GraphBuilderContext::buffer ( )
inline

Definition at line 154 of file Context.h.

154{ return _buffer_context; }

Referenced by tflimport::Conv2DGraphBuilder::build(), tflimport::DepthwiseConv2DGraphBuilder::build(), and tflimport::copy_tensors().

◆ d()

coco::Data * tflimport::GraphBuilderContext::d ( )
inline

◆ graph()

◆ m()

◆ tensor()


The documentation for this class was generated from the following file: