Packages

tflite_beam

0.3.2
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 error_reporter.h
Raw

c_src/tflite/error_reporter.h 

#ifndef TFLITE_ERROR_REPORTER_BINDINGS_H

#define TFLITE_ERROR_REPORTER_BINDINGS_H



#pragma once



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

#include "../erlang_nif_resource.h"



// --------------------- nif api --------------------



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



// ------------------ internal api ------------------



/// @brief Get error reporter

/// @param env [in] NIF env 

/// @param error_reporter_term [in] NIF term that stores the error reporter resource or `nil`

/// @param error_reporter_res [out] resource

/// @param error_reporter [out] pointer to the error reporter

/// @param error_term [out] an error tuple will be set if any error occurred

/// @return true if sucessfully get the error reporter, otherwise false

bool _get_error_reporter(ErlNifEnv *env, ERL_NIF_TERM error_reporter_term, NifResErrorReporter *&error_reporter_res, tflite::ErrorReporter * &error_reporter, ERL_NIF_TERM &error_term);



NifResErrorReporter * _make_error_reporter(ErlNifEnv *env, tflite::ErrorReporter * e, ERL_NIF_TERM &out);



#endif // TFLITE_VERIFIER_BINDINGS_H