Autofill for SavedModel

pangge · Nov 14, 2018 · ea9b6ed · ea9b6ed
1 parent fadfbbe
commit ea9b6ed
Show file tree

Hide file tree

Showing 16 changed files with 857 additions and 143 deletions.
diff --git a/src/core/BUILD b/src/core/BUILD
@@ -120,6 +120,21 @@ serving_proto_library_py(
     ],
 )
 
+cc_library(
+    name = "autofill",
+    srcs = ["autofill.cc"],
+    hdrs = ["autofill.h"],
+    deps = [
+        ":constants",
+        ":logging",
+        ":model_config",
+        ":model_config_proto",
+        "@org_tensorflow//tensorflow/cc/saved_model:tag_constants",
+        "@org_tensorflow//tensorflow/c:c_api",
+        "@org_tensorflow//tensorflow/core:lib",
+    ],
+)
+
 cc_library(
     name = "constants",
     hdrs = ["constants.h"],
@@ -299,7 +314,9 @@ cc_library(
     srcs = ["utils.cc"],
     hdrs = ["utils.h"],
     deps = [
+        ":autofill",
         ":constants",
+        ":logging",
         ":model_config_proto",
         "@org_tensorflow//tensorflow/c:c_api",
         "@org_tensorflow//tensorflow/core:lib",