diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-07 15:41:22 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-07 17:27:20 -0700 |
| commit | fc7f0b296dd53d1b72af21d36d36b6bcc5291ea7 (patch) | |
| tree | 46e76ead2391a3fb1232459189ad0b8d0d8066ac /tensorflow/contrib/lite/kernels/BUILD | |
| parent | 3a2f1cfb73fa6a21eba077485bdc08aa05646ad1 (diff) | |
Add support for select (via tf.where) support to tflite.
PiperOrigin-RevId: 195734246
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index feab18b5c2..79e3c9f266 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -164,6 +164,7 @@ cc_library( "register.cc", "reshape.cc", "resize_bilinear.cc", + "select.cc", "skip_gram.cc", "space_to_batch_nd.cc", "space_to_depth.cc", @@ -870,6 +871,23 @@ tf_cc_test( ], ) +tf_cc_test( + name = "select_test", + size = "small", + srcs = [ + "select_test.cc", + ], + tags = [ + "tflite_not_portable_ios", + ], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + filegroup( name = "all_files", srcs = glob( |