Decode Tensorflow operator list

Author: wedesoft

aiscm/Makefile.am

@@ -28,7 +28,7 @@ AISCM_LDFLAGS = -no-undefined -module -shared -avoid-version
 tensorflow_generate_SOURCES = tensorflow-generate.c
 tensorflow_generate_CFLAGS = -I/usr/local/include/ClearSilver -I/usr/include/ClearSilver
 tensorflow_generate_LDFLAGS =
-tensorflow_generate_LDADD = -lneo_cs -lneo_utl
+tensorflow_generate_LDADD = ../tensorflow/core/framework/libtensorflow-ops.la -lneo_cs -lneo_utl -ltensorflow
 
 libguile_aiscm_tests_la_SOURCES = tests.c ffmpeg-helpers.c ffmpeg-helpers-tests.c ringbuffer.c ringbuffer-tests.c \
 																	util-helpers.c util-helpers-tests.c samples-helpers.c samples-helpers-tests.c

aiscm/tensorflow-generate.c

@@ -1,5 +1,8 @@
 #include <stdio.h>
 #include <ClearSilver.h>
+#include <tensorflow/c/c_api.h>
+#include "tensorflow/core/framework/op_def.pb-c.h"
+
 
 NEOERR *output(void *ctx, char *text)
 {
@@ -9,6 +12,9 @@ NEOERR *output(void *ctx, char *text)
 
 int main(void)
 {
+  TF_Buffer *buffer = TF_GetAllOpList();
+  Tensorflow__OpList *op_list = tensorflow__op_list__unpack(NULL, buffer->length, buffer->data);
+
   HDF *hdf;
   hdf_init(&hdf);
 
@@ -40,4 +46,7 @@ int main(void)
 
   cs_destroy(&parse);
   hdf_destroy(&hdf);
+
+  tensorflow__op_list__free_unpacked(op_list, NULL);
+  TF_DeleteBuffer(buffer);
 }

configure.ac

@@ -176,6 +176,16 @@ AC_TRY_LINK([#include <ClearSilver.h>],
 LIBS="$old_LIBS"
 CFLAGS="$old_CFLAGS"
 
+old_LIBS="$LIBS"
+LIBS="$LIBS -lprotobuf-c"
+AC_MSG_CHECKING([for Protocol Buffer library])
+AC_TRY_LINK([#include <protobuf-c/protobuf-c.h>],
+            [protobuf_c_empty_string],
+            [AC_MSG_RESULT([yes])],
+            [AC_MSG_RESULT([no])]
+            [AC_MSG_ERROR([Could not locate Protocol Buffer library])])
+LIBS="$old_LIBS"
+
 old_LIBS="$LIBS"
 LIBS="$LIBS -ltensorflow"
 AC_MSG_CHECKING([for Tensorflow library])

tensorflow/core/framework/Makefile.am

@@ -10,4 +10,4 @@ noinst_LTLIBRARIES = libtensorflow-ops.la
 
 libtensorflow_ops_la_SOURCES = attr_value.pb-c.c op_def.pb-c.c resource_handle.pb-c.c tensor.pb-c.c tensor_shape.pb-c.c types.pb-c.c
 libtensorflow_ops_la_CFLAGS =
-libtensorflow_ops_la_LIBADD =
+libtensorflow_ops_la_LIBADD = -lprotobuf-c