filter_tensorflow: new Tensorflow Lite filter plugin! (#2511)

Signed-off-by: Masoud Koleini <masoud.koleini@arm.com>
fluent · Oct 7, 2020 · 5c054aa · 5c054aa
1 parent b8053b7
commit 5c054aa
Show file tree

Hide file tree

Showing 5 changed files with 532 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -191,6 +191,7 @@ option(FLB_FILTER_THROTTLE_SIZE "Enable throttle size filter"        No)
 option(FLB_FILTER_NEST        "Enable nest filter"                  Yes)
 option(FLB_FILTER_LUA         "Enable Lua scripting filter"         Yes)
 option(FLB_FILTER_RECORD_MODIFIER "Enable record_modifier filter"   Yes)
+option(FLB_FILTER_TENSORFLOW  "Enable tensorflow filter"             No)
 
 # Debug callbacks
 option(FLB_HTTP_CLIENT_DEBUG  "Enable HTTP Client debug callbacks"   No)

diff --git a/plugins/CMakeLists.txt b/plugins/CMakeLists.txt
@@ -209,6 +209,7 @@ REGISTER_FILTER_PLUGIN("filter_aws")
 REGISTER_FILTER_PLUGIN("filter_record_modifier")
 REGISTER_FILTER_PLUGIN("filter_throttle")
 REGISTER_FILTER_PLUGIN("filter_throttle_size")
+REGISTER_FILTER_PLUGIN("filter_tensorflow")
 
 if(FLB_REGEX)
   REGISTER_FILTER_PLUGIN("filter_kubernetes")

diff --git a/plugins/filter_tensorflow/CMakeLists.txt b/plugins/filter_tensorflow/CMakeLists.txt
@@ -0,0 +1,6 @@
+set(src
+  tensorflow.c)
+
+FLB_PLUGIN(filter_tensorflow "${src}" "")
+target_include_directories(flb-plugin-filter_tensorflow PRIVATE ${Tensorflow_DIR})
+target_link_libraries(flb-plugin-filter_tensorflow -ltensorflowlite_c)