tflimport::TflBufferContext Class Reference

Class to read and provide buffer information of tflite. More...

#include <Context.h>

Data Structures
struct	TflBuffer

Public Member Functions
	TflBufferContext (const tflite::Model *tfl_model)
template<typename T >
TflBuffer< T >	tensor_buffer (const tflite::SubGraph *graph, uint32_t tensor_idx) const

Detailed Description

Class to read and provide buffer information of tflite.

Definition at line 86 of file Context.h.

Constructor & Destructor Documentation

TflBufferContext()

tflimport::TflBufferContext::TflBufferContext ( const tflite::Model *  tfl_model )

explicit

Definition at line 104 of file Context.cpp.

{
  const flatbuffers::Vector<flatbuffers::Offset<tflite::Buffer>> *tfl_buffers;
 
  tfl_buffers = tfl_model->buffers();
 
  for (uint32_t buffer_id = 0; buffer_id < tfl_buffers->size(); ++buffer_id)
  {
    _buffer_ctx[buffer_id] = (*tfl_buffers)[buffer_id];
  }
}

References flatbuffers::Vector< T >::size().

Member Function Documentation

tensor_buffer()

template<typename T >

TflBuffer< T > tflimport::TflBufferContext::tensor_buffer ( const tflite::SubGraph *  graph, uint32_t  tensor_idx  ) const

inline

Definition at line 101 of file Context.h.

  {
    TflBufferContext::TflBuffer<T> res{nullptr, 0};
    const auto *tensor = graph->tensors()->Get(tensor_idx);
    uint32_t tfl_buf_id = tensor->buffer();
 
    assert(_buffer_ctx.size() > tfl_buf_id);
 
    const tflite::Buffer *tfl_buffer = _buffer_ctx.at(tfl_buf_id);
 
    if (auto *array = tfl_buffer->data())
    {
      if (size_t size = array->size())
      {
        assert(size % sizeof(T) == 0);
 
        res.len = size / sizeof(T);
        res.ptr = reinterpret_cast<const T *>(array->data());
      }
    }
 
    return res;
  }

References size.

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

---

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

• compiler/enco/frontend/tflite/src/Context.h
• compiler/enco/frontend/tflite/src/Context.cpp