Packages

tflite_beam

0.2.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
3 Dependencies
Dependants
Files
Activity
tflite_beam c_src tflite ops builtin builtin_resolver.cpp
Raw

c_src/tflite/ops/builtin/builtin_resolver.cpp 

#include <erl_nif.h>

#include "../../../nif_utils.hpp"

#include "../../../erlang_nif_resource.h"

#include "../../../helper.h"



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

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



#include "builtin_resolver.h"



ERL_NIF_TERM ops_builtin_builtin_resolver_new(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]) {

    NifResBuiltinOpResolver * res = nullptr;

    ERL_NIF_TERM ret;



    if (!(res = NifResBuiltinOpResolver::allocate_resource(env, ret))) {

        return ret;

    }



    res->val = new tflite::ops::builtin::BuiltinOpResolver();

    ret = enif_make_resource(env, res);

    enif_release_resource(res);

    return erlang::nif::ok(env, ret);

}