Remove prefix from the graph API

Author: lantiga

src/backends/tensorflow.c

@@ -136,13 +136,11 @@ TF_Tensor* RAI_TFTensorFromTensor(RAI_Tensor* t){
 }
 
 
-RAI_Graph *RAI_GraphCreateTF(const char *prefix,
-                             RAI_Backend backend, RAI_Device device,
+RAI_Graph *RAI_GraphCreateTF(RAI_Backend backend, RAI_Device device,
                              const char *graphdef, size_t graphlen) {
   TF_Graph* graph = TF_NewGraph();
 
   TF_ImportGraphDefOptions* options = TF_NewImportGraphDefOptions();
-  TF_ImportGraphDefOptionsSetPrefix(options, prefix);
 
   TF_Buffer *buffer = TF_NewBuffer();
   buffer->length = graphlen;

src/backends/tensorflow.h

@@ -11,8 +11,7 @@ RAI_Tensor* RAI_TensorCreateFromTFTensor(TF_Tensor *tensor);
 
 TF_Tensor* RAI_TFTensorFromTensor(RAI_Tensor* t);
 
-RAI_Graph *RAI_GraphCreateTF(const char *prefix,
-                             RAI_Backend backend, RAI_Device device,
+RAI_Graph *RAI_GraphCreateTF(RAI_Backend backend, RAI_Device device,
                              const char *graphdef, size_t graphlen);
 
 void RAI_GraphFreeTF(RAI_Graph* graph);

src/graph.c

@@ -37,11 +37,10 @@ int RAI_GraphInit(RedisModuleCtx* ctx) {
   return RedisAI_GraphType != NULL;
 }
 
-RAI_Graph *RAI_GraphCreate(const char *prefix,
-                           RAI_Backend backend, RAI_Device device,
+RAI_Graph *RAI_GraphCreate(RAI_Backend backend, RAI_Device device,
                            const char *graphdef, size_t graphlen) {
   if (backend == RAI_BACKEND_TENSORFLOW) {
-    return RAI_GraphCreateTF(prefix, backend, device, graphdef, graphlen);
+    return RAI_GraphCreateTF(backend, device, graphdef, graphlen);
   }
 
   return NULL;

src/graph.h

@@ -16,7 +16,7 @@
 extern RedisModuleType *RedisAI_GraphType;
 
 int RAI_GraphInit(RedisModuleCtx* ctx);
-RAI_Graph* RAI_GraphCreate(const char* prefix, RAI_Backend backend, RAI_Device device, const char* graphdef, size_t graphlen);
+RAI_Graph* RAI_GraphCreate(RAI_Backend backend, RAI_Device device, const char* graphdef, size_t graphlen);
 void RAI_GraphFree(RAI_Graph* graph);
 RAI_GraphRunCtx* RAI_RunCtxCreate(RAI_Graph* graph);
 int RAI_RunCtxAddInput(RAI_GraphRunCtx* gctx, const char* inputName, RAI_Tensor* inputTensor);

src/redisai.c

@@ -388,7 +388,7 @@ int RedisAI_Get_Tensor_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **arg
 
 // ================================
 
-// key graphbuf [prefix]
+// key graphbuf
 int RedisAI_Set_Graph_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
   RedisModule_AutoMemory(ctx);
 
@@ -427,12 +427,8 @@ int RedisAI_Set_Graph_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv
 
   size_t graphlen;
   const char *graphdef = RedisModule_StringPtrLen(argv[4], &graphlen);
-  const char *prefix = "";
-  if (argc == 6) {
-    const char *prefix = RedisModule_StringPtrLen(argv[5], NULL);
-  }
 
-  graph = RAI_GraphCreate(prefix, backend, device, graphdef, graphlen);
+  graph = RAI_GraphCreate(backend, device, graphdef, graphlen);
 
   if(graph == NULL){
     return RedisModule_ReplyWithError(ctx, "ERR failed creating the graph");

src/redisai.h

@@ -31,7 +31,7 @@ long long MODULE_API_FUNC(RedisAI_TensorDim)(RAI_Tensor* t, int dim);
 size_t MODULE_API_FUNC(RedisAI_TensorByteSize)(RAI_Tensor* t);
 char* MODULE_API_FUNC(RedisAI_TensorData)(RAI_Tensor* t);
 
-RAI_Graph* MODULE_API_FUNC(RedisAI_GraphCreate)(const char* prefix, RAI_Backend backend, RAI_Device device, const char* graphdef, size_t graphlen);
+RAI_Graph* MODULE_API_FUNC(RedisAI_GraphCreate)(RAI_Backend backend, RAI_Device device, const char* graphdef, size_t graphlen);
 void MODULE_API_FUNC(RedisAI_GraphFree)(RAI_Graph* graph);
 RAI_GraphRunCtx* MODULE_API_FUNC(RedisAI_RunCtxCreate)(RAI_Graph* graph);
 int MODULE_API_FUNC(RedisAI_RunCtxAddInput)(RAI_GraphRunCtx* gctx, const char* inputName, RAI_Tensor* inputTensor);