TensorFlow 2.0 Beta is available Learn more

tflite

Deserialization infrastructure for tflite.

Summary

Provides functionality to go from a serialized tflite model in flatbuffer format to an interpreter.

Functions

GetRegistrationFromOpCode(const OperatorCode *opcode, const OpResolver & op_resolver, ErrorReporter *error_reporter, const TfLiteRegistration **registration)
TfLiteStatus
typeToTfLiteType()
constexpr TfLiteType
typeToTfLiteType< TfLiteFloat16 >()
constexpr TfLiteType
typeToTfLiteType< bool >()
constexpr TfLiteType
typeToTfLiteType< float >()
constexpr TfLiteType
typeToTfLiteType< int >()
constexpr TfLiteType
typeToTfLiteType< int16_t >()
constexpr TfLiteType
typeToTfLiteType< int64_t >()
constexpr TfLiteType
typeToTfLiteType< int8_t >()
constexpr TfLiteType
typeToTfLiteType< std::complex< float > >()
constexpr TfLiteType
typeToTfLiteType< string >()
constexpr TfLiteType
typeToTfLiteType< unsigned char >()
constexpr TfLiteType

Classes

tflite::ErrorReporter

A functor that reports error to supporting system.

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::OpResolver

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

tflite::TfLiteVerifier

Abstract interface that verifies whether a given model is legit.

Functions

GetRegistrationFromOpCode

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

typeToTfLiteType

constexpr TfLiteType typeToTfLiteType()

typeToTfLiteType< TfLiteFloat16 >

constexpr TfLiteType typeToTfLiteType< TfLiteFloat16 >()

typeToTfLiteType< bool >

constexpr TfLiteType typeToTfLiteType< bool >()

typeToTfLiteType< float >

constexpr TfLiteType typeToTfLiteType< float >()

typeToTfLiteType< int >

constexpr TfLiteType typeToTfLiteType< int >()

typeToTfLiteType< int16_t >

constexpr TfLiteType typeToTfLiteType< int16_t >()

typeToTfLiteType< int64_t >

constexpr TfLiteType typeToTfLiteType< int64_t >()

typeToTfLiteType< int8_t >

constexpr TfLiteType typeToTfLiteType< int8_t >()

typeToTfLiteType< std::complex< float > >

constexpr TfLiteType typeToTfLiteType< std::complex< float > >()

typeToTfLiteType< string >

constexpr TfLiteType typeToTfLiteType< string >()

typeToTfLiteType< unsigned char >

constexpr TfLiteType typeToTfLiteType< unsigned char >()