Packages

tflite_beam

0.3.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
0 Dependencies
Dependants
Files
Activity
tflite_beam c_src helper.h
Raw

c_src/helper.h 

#ifndef TFLITE_HELPER_H

#define TFLITE_HELPER_H



#pragma once



#include <erl_nif.h>

#include "erlang_nif_resource.h"



bool tensor_type_to_erl_term(const TfLiteType in_type, ErlNifEnv *env, ERL_NIF_TERM &out_term);

bool tensor_type_from_erl_term(ErlNifEnv *env, const ERL_NIF_TERM in_term, TfLiteType &out_type);



#endif  // TFLITE_HELPER_H