Packages

tflite_elixir

0.1.5
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 erlang_nif_resource.hpp
Raw

c_src/erlang_nif_resource.hpp 

#ifndef TFLITE_ELIXIR_ERLANG_NIF_RESOURCE_HPP

#define TFLITE_ELIXIR_ERLANG_NIF_RESOURCE_HPP



#pragma once



#include <atomic>

#include <erl_nif.h>



#include "tensorflow/lite/c/c_api.h"

#include "tensorflow/lite/c/common.h"

#include "tensorflow/lite/core/api/error_reporter.h"

#include "tensorflow/lite/interpreter_builder.h"

#include "tensorflow/lite/interpreter.h"

#include "tensorflow/lite/kernels/register.h"

#include "tensorflow/lite/kernels/builtin_op_kernels.h"

#include "tensorflow/lite/model.h"

#include "tensorflow/lite/stderr_reporter.h"



struct NifResBuiltinOpResolver {

    tflite::ops::builtin::BuiltinOpResolver * val;

    // OpResolver must live as long as its InterpreterBuilder

    std::atomic_int64_t reference_count{0};

    std::atomic_bool dropped_in_erlang{false};

    std::atomic_bool deleted{false};

    static ErlNifResourceType * type;

};



struct NifResFlatBufferModel {

    tflite::FlatBufferModel * val;

    // FlatBufferModel must live as long as its Interpreter

    std::atomic_int64_t reference_count{0};

    std::atomic_bool dropped_in_erlang{false};

    std::atomic_bool deleted{false};

    static ErlNifResourceType * type;

};



struct NifResInterpreterBuilder {

    tflite::InterpreterBuilder * val;

    NifResBuiltinOpResolver * op_resolver;

    NifResFlatBufferModel * flatbuffer_model;

    static ErlNifResourceType * type;

};



struct NifResInterpreter {

    tflite::Interpreter * val;

    NifResFlatBufferModel * flatbuffer_model;

    static ErlNifResourceType * type;

};



struct NifResErrorReporter {

    tflite::ErrorReporter * val;

    std::atomic_bool is_default{false};

    static ErlNifResourceType * type;

};



struct NifResTfLiteTensor {

    TfLiteTensor * val;

    std::atomic_bool borrowed{false};

    static ErlNifResourceType * type;

};



#ifdef CORAL_SUPPORT_ENABLED



#include "tflite/public/edgetpu.h"



struct NifResEdgeTpuContext {

    edgetpu::EdgeTpuContext * val;

    static ErlNifResourceType * type;

};



#endif



#endif //TFLITE_ELIXIR_ERLANG_NIF_RESOURCE_HPP