aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/schema
diff options
context:
space:
mode:
authorGravatar Andrew Selle <aselle@google.com>2018-09-04 15:57:55 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-04 16:02:36 -0700
commit69753ba5dbe5950639efc1b5e065901651cd8973 (patch)
tree5d143d1ac3c135730024e132c54b626bff6e072f /tensorflow/contrib/lite/schema
parent5cb997a35383bc2832be5a415d72aa950374ebfa (diff)
Create a way to serialize Interpreter data to a flatbuffer.
PiperOrigin-RevId: 211540844
Diffstat (limited to 'tensorflow/contrib/lite/schema')
-rw-r--r--tensorflow/contrib/lite/schema/BUILD14
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/schema/BUILD b/tensorflow/contrib/lite/schema/BUILD
index 28a7e50003..55bf2c48b9 100644
--- a/tensorflow/contrib/lite/schema/BUILD
+++ b/tensorflow/contrib/lite/schema/BUILD
@@ -56,6 +56,20 @@ flatbuffer_cc_library(
srcs = ["schema.fbs"],
)
+# Generic schema for inference on device (but with reflections makes bigger).
+flatbuffer_cc_library(
+ name = "schema_fbs_with_reflection",
+ srcs = ["schema.fbs"],
+ flatc_args = [
+ "--reflect-types",
+ "--reflect-names",
+ "--no-union-value-namespacing",
+ "--gen-object-api",
+ ],
+ gen_reflections = True,
+ out_prefix = "reflection/",
+)
+
# Schema test to make sure we don't introduce backward incompatible changes
# to schemas.
cc_test(