fixed tflite files

Author: DvirDukhan

src/backends/tflite.c

@@ -48,12 +48,12 @@ RAI_Model *RAI_ModelCreateTFLite(RAI_Backend backend, const char *devicestr, RAI
         return NULL;
     }
 
-    size_t ninputs = tfliteModelNumInputs(model, &error_descr, RedisModule_Alloc);
+    size_t ninputs = tfliteModelNumInputs(model, &error_descr);
     if (error_descr) {
         goto cleanup;
     }
 
-    size_t noutputs = tfliteModelNumOutputs(model, &error_descr, RedisModule_Alloc);
+    size_t noutputs = tfliteModelNumOutputs(model, &error_descr);
     if (error_descr) {
         goto cleanup;
     }
@@ -62,16 +62,15 @@ RAI_Model *RAI_ModelCreateTFLite(RAI_Backend backend, const char *devicestr, RAI
     outputs_ = array_new(char *, noutputs);
 
     for (size_t i = 0; i < ninputs; i++) {
-        const char *input = tfliteModelInputNameAtIndex(model, i, &error_descr, RedisModule_Alloc);
+        const char *input = tfliteModelInputNameAtIndex(model, i, &error_descr);
         if (error_descr) {
             goto cleanup;
         }
         inputs_ = array_append(inputs_, RedisModule_Strdup(input));
     }
 
     for (size_t i = 0; i < noutputs; i++) {
-        const char *output =
-            tfliteModelOutputNameAtIndex(model, i, &error_descr, RedisModule_Alloc);
+        const char *output = tfliteModelOutputNameAtIndex(model, i, &error_descr);
         ;
         if (error_descr) {
             goto cleanup;

src/libtflite_c/tflite_c.cpp

@@ -263,52 +263,52 @@ extern "C" void *tfliteLoadModel(const char *graph, size_t graphlen, DLDeviceTyp
     return ctx;
 }
 
-extern "C" size_t tfliteModelNumInputs(void* ctx, char** error, void *(*alloc)(size_t)) {
+extern "C" size_t tfliteModelNumInputs(void* ctx, char** error) {
     ModelContext *ctx_ = (ModelContext*) ctx;
     size_t ret = 0;
     try {
         auto interpreter = ctx_->interpreter;
         ret =  interpreter->inputs().size();
     }
     catch(std::exception ex) {
-        setError(ex.what(), error, alloc);
+        _setError(ex.what(), error);
     }
     return ret;
 }
 
-extern "C" const char* tfliteModelInputNameAtIndex(void* modelCtx, size_t index, char** error, void *(*alloc)(size_t)) {
+extern "C" const char* tfliteModelInputNameAtIndex(void* modelCtx, size_t index, char** error) {
     ModelContext *ctx_ = (ModelContext*) modelCtx;
     const char* ret = NULL;
     try {
         ret = ctx_->interpreter->GetInputName(index);
     }
     catch(std::exception ex) {
-        setError(ex.what(), error, alloc);
+        _setError(ex.what(), error);
     }
     return ret;
 }
 
-extern "C" size_t tfliteModelNumOutputs(void* ctx, char** error, void *(*alloc)(size_t)) {
+extern "C" size_t tfliteModelNumOutputs(void* ctx, char** error) {
     ModelContext *ctx_ = (ModelContext*) ctx;
     size_t ret = 0;
     try {
         auto interpreter = ctx_->interpreter;
         ret =  interpreter->outputs().size();
     }
     catch(std::exception ex) {
-        setError(ex.what(), error, alloc);
+        _setError(ex.what(), error);
     }
     return ret;
 }
 
-extern "C" const char* tfliteModelOutputNameAtIndex(void* modelCtx, size_t index, char** error, void *(*alloc)(size_t)) {
+extern "C" const char* tfliteModelOutputNameAtIndex(void* modelCtx, size_t index, char** error) {
     ModelContext *ctx_ = (ModelContext*) modelCtx;
     const char* ret = NULL;
     try {
         ret = ctx_->interpreter->GetOutputName(index);
     }
     catch(std::exception ex) {
-        setError(ex.what(), error, alloc);
+        _setError(ex.what(), error);
     }
     return ret;
 }

src/libtflite_c/tflite_c.h

@@ -19,15 +19,13 @@ void tfliteSerializeModel(void *ctx, char **buffer, size_t *len, char **error);
 
 void tfliteDeallocContext(void *ctx);
 
-size_t tfliteModelNumInputs(void *ctx, char **error, void *(*alloc)(size_t));
+size_t tfliteModelNumInputs(void *ctx, char **error);
 
-const char *tfliteModelInputNameAtIndex(void *modelCtx, size_t index, char **error,
-                                        void *(*alloc)(size_t));
+const char *tfliteModelInputNameAtIndex(void *modelCtx, size_t index, char **error);
 
-size_t tfliteModelNumOutputs(void *ctx, char **error, void *(*alloc)(size_t));
+size_t tfliteModelNumOutputs(void *ctx, char **error);
 
-const char *tfliteModelOutputNameAtIndex(void *modelCtx, size_t index, char **error,
-                                         void *(*alloc)(size_t));
+const char *tfliteModelOutputNameAtIndex(void *modelCtx, size_t index, char **error);
 
 #ifdef __cplusplus
 }