Packages

tflite_beam

0.3.1
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
0 Dependencies
Dependants
Files
Activity
tflite_beam c_src tflite interpreter_builder.h
Raw

c_src/tflite/interpreter_builder.h 

#ifndef TFLITE_INTERPRETER_BUILDER_BINDINGS_H

#define TFLITE_INTERPRETER_BUILDER_BINDINGS_H



#pragma once



#include <erl_nif.h>



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

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

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



#endif // TFLITE_INTERPRETER_BUILDER_BINDINGS_H