Packages

tflite_elixir

0.1.6
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.0 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2

TensorFlow Lite Elixir binding with optional TPU support.

HexDocs

Current section

Files

Jump to
Readme
15 Versions
3 Dependencies
Dependants
Files
Activity
tflite_elixir 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 flatBufferModel_buildFromFile(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);



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



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



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



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



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



ERL_NIF_TERM flatBufferModel_readAllMetadata(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

/// @return out nif resource

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



#endif // TFLITE_FLATBUFFERMODEL_BINDINGS_H