diff options
| author |  Jared Duke <jdduke@google.com> | 2018-07-23 14:49:59 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-23 15:05:12 -0700 |
| commit | 218bd6facc481da1ed199a8c680427051cb1c6cb (patch) | |
| tree | c15055ac69a414c55673a2c686c18fb7af60c3c9 /tensorflow/contrib/lite/build_def.bzl | |
| parent | cf94a46c34f8568608d78b77e9a1c4369ebcafa2 (diff) | |
Add initial experimental C API for TFLite
PiperOrigin-RevId: 205728711
Diffstat (limited to 'tensorflow/contrib/lite/build_def.bzl')
| -rw-r--r-- | tensorflow/contrib/lite/build_def.bzl | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/build_def.bzl b/tensorflow/contrib/lite/build_def.bzl index bed862454e..79f7455ad8 100644 --- a/tensorflow/contrib/lite/build_def.bzl +++ b/tensorflow/contrib/lite/build_def.bzl @@ -2,6 +2,7 @@ load( "//tensorflow:tensorflow.bzl", "tf_cc_test", + "tf_cc_shared_object", ) def tflite_copts(): @@ -118,6 +119,20 @@ def tflite_jni_binary(name, deps= deps + [linkscript], linkopts=linkopts) +def tflite_cc_shared_object(name, + copts=tflite_copts(), + linkopts=[], + linkstatic=1, + deps=[]): + """Builds a shared object for TFLite.""" + tf_cc_shared_object( + name=name, + copts=copts, + linkstatic=linkstatic, + linkopts=linkopts + tflite_jni_linkopts(), + framework_so=[], + deps=deps) + def tf_to_tflite(name, src, options, out): """Convert a frozen tensorflow graphdef to TF Lite's flatbuffer. |