Packages

tflite_beam

0.2.0
0.3.10 0.3.9 0.3.8 0.3.7 0.3.6 0.3.5 0.3.4 0.3.3 0.3.2 0.3.1 0.3.0 0.2.1 0.2.0

TensorFlow Lite BEAM binding with optional TPU support.

HexDocs

Current section

Files

Jump to
Readme
13 Versions
3 Dependencies
Dependants
Files
Activity
tflite_beam c_src tflite flatbuffer_model.h
Raw

c_src/tflite/flatbuffer_model.h 

#ifndef TFLITE_FLATBUFFERMODEL_BINDINGS_H

#define TFLITE_FLATBUFFERMODEL_BINDINGS_H



#pragma once



// --------------------- nif api --------------------



ERL_NIF_TERM flatbuffer_model_build_from_file(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_verify_and_build_from_file(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_build_from_buffer(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_initialized(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_error_reporter(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_get_minimum_runtime(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);

ERL_NIF_TERM flatbuffer_model_read_all_metadata(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);



// ------------------ internal api ------------------



/// @brief Make flatbuffer model nif resource

/// @param env Nif env

/// @param m [in] reference to FlatBufferModel, model ownership will be transferred to res

/// @param out out nif resource term

/// @param copied_buffer copied_buffer

/// @return out nif resource

NifResFlatBufferModel * _make_flatbuffer_model_resource(ErlNifEnv *env, std::unique_ptr<tflite::FlatBufferModel>& m, ERL_NIF_TERM &out, void * copied_buffer = nullptr);



#endif // TFLITE_FLATBUFFERMODEL_BINDINGS_H