ONE - On-device Neural Engine
Loading...
Searching...
No Matches
nnkit::support::tflite::InputTensorSet Class Referencefinal

#include <TensorSets.h>

Collaboration diagram for nnkit::support::tflite::InputTensorSet:

Public Member Functions

 InputTensorSet (::tflite::Interpreter &interp)
 
uint32_t size (void) const override
 
TfLiteTensor * at (uint32_t n) const override
 
- Public Member Functions inherited from nnkit::support::tflite::TensorSet
virtual ~TensorSet ()=default
 

Detailed Description

Definition at line 31 of file TensorSets.h.

Constructor & Destructor Documentation

◆ InputTensorSet()

nnkit::support::tflite::InputTensorSet::InputTensorSet ( ::tflite::Interpreter &  interp)
inlineexplicit

Definition at line 34 of file TensorSets.h.

34 : _interp(interp)
35 {
36 // DO NOTHING
37 }

Member Function Documentation

◆ at()

TfLiteTensor * nnkit::support::tflite::InputTensorSet::at ( uint32_t  n) const
inlineoverridevirtual

Implements nnkit::support::tflite::TensorSet.

Definition at line 43 of file TensorSets.h.

43{ return _interp.tensor(_interp.inputs().at(n)); }

◆ size()

uint32_t nnkit::support::tflite::InputTensorSet::size ( void  ) const
inlineoverridevirtual

Implements nnkit::support::tflite::TensorSet.

Definition at line 40 of file TensorSets.h.

40{ return _interp.inputs().size(); }

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