Stay organized with collections Save and categorize content based on your preferences.

tflite

\file Memory management for TF Lite.

Summary

\file Deserialization infrastructure for tflite.

\file Provides functionality to construct an interpreter for a model.

\file Main abstraction controlling the tflite interpreter.

\file This provides a few C++ helpers that are useful for manipulating C structures in C++.

Do NOT include this file directly, instead include third_party/tensorflow/lite/interpreter.h See third_party/tensorflow/lite/c/common.h for the API for defining operations (TfLiteRegistration).

Provides functionality to go from a serialized tflite model in flatbuffer format to an in-memory representation of the model.

Functions
DefaultErrorReporter()
GetRegistrationFromOpCode(const OperatorCode *opcode, const OpResolver & op_resolver, ErrorReporter *error_reporter, const TfLiteRegistration **registration)
TfLiteStatus

Classes
tflite::Allocation

A memory allocation handle. This could be a mmap or shared memory.

tflite::ErrorReporter

A functor that reports error to supporting system.

tflite::FileCopyAllocation
tflite::FlatBufferModel

An RAII object that represents a read-only tflite model, copied from disk, or mmapped.

tflite::Interpreter

An interpreter for a graph of nodes that input and output from tensors.

tflite::InterpreterBuilder

Build an interpreter capable of interpreting model.

tflite::MMAPAllocation

Note that not all platforms support MMAP-based allocation.

tflite::MemoryAllocation
tflite::MutableOpResolver

An OpResolver that is mutable, also used as the op in gen_op_registration.

tflite::OpResolver

Abstract interface that returns TfLiteRegistrations given op codes or custom op names.

tflite::TfLiteIntArrayView

Provides a range iterable wrapper for TfLiteIntArray* (C lists) that TfLite C api uses.

Structs
tflite::StderrReporter

Namespaces
tflite::delegates
tflite::interpreter_wrapper
tflite::op_resolver_hasher

Functions

DefaultErrorReporter

ErrorReporter * DefaultErrorReporter()

GetRegistrationFromOpCode

TfLiteStatus GetRegistrationFromOpCode(
  const OperatorCode *opcode,
  const OpResolver & op_resolver,
  ErrorReporter *error_reporter,
  const TfLiteRegistration **registration
)