nnkit::support::tflite::OutputTensorSet Class Reference

#include <TensorSets.h>

Collaboration diagram for nnkit::support::tflite::OutputTensorSet:

Public Member Functions
	OutputTensorSet (::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 49 of file TensorSets.h.

Constructor & Destructor Documentation

OutputTensorSet()

nnkit::support::tflite::OutputTensorSet::OutputTensorSet ( ::tflite::Interpreter &  interp )

inline

Definition at line 52 of file TensorSets.h.

                                             : _interp(interp)
  {
    // DO NOTHING
  }

Member Function Documentation

at()

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

inlineoverridevirtual

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

Definition at line 61 of file TensorSets.h.

{ return _interp.tensor(_interp.outputs().at(n)); }

size()

uint32_t nnkit::support::tflite::OutputTensorSet::size ( void  ) const

inlineoverridevirtual

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

Definition at line 58 of file TensorSets.h.

{ return _interp.outputs().size(); }

---

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

• compiler/nnkit-tflite/support/include/nnkit/support/tflite/TensorSets.h