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 coral coral.h
Raw

c_src/coral/coral.h 

#ifndef TFLITE_CORAL_BINDINGS_H

#define TFLITE_CORAL_BINDINGS_H



#include <vector>

#include <string.h>

#include <erl_nif.h>

#include "../nif_utils.hpp"



#include "tflite/public/edgetpu.h"

#include "tensorflow/lite/interpreter.h"

#include "tensorflow/lite/model.h"



#pragma once



void destruct_egdetpu_context(ErlNifEnv *env, void *args);



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



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



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



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



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



#endif // TFLITE_CORAL_BINDINGS_H