diff options
author | Alan Chiao <alanchiao@google.com> | 2018-09-04 20:22:55 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 20:26:56 -0700 |
commit | e9332539bea372f6dbe6ef185f9d8b1f3b6e1fe2 (patch) | |
tree | c7d5490402934bc34b6ad76a89ac14233e91a848 /tensorflow/contrib/lite/kernels/BUILD | |
parent | ecb6bc19e0cdbd2f2e98de909b4f3b8ca9fd7ab1 (diff) |
Relu1 custom op.
This is implemented as custom op instead of builtin op because Relu1 is not
supported in Tensorflow and not commonly used.
PiperOrigin-RevId: 211571619
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 ab989c5425..b7c5cbf207 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -192,6 +192,7 @@ cc_library( "pooling.cc", "pow.cc", "reduce.cc", + "relu1.cc", "reshape.cc", "resize_bilinear.cc", "select.cc", @@ -305,6 +306,23 @@ tf_cc_test( ) tf_cc_test( + name = "relu1_test", + size = "small", + srcs = ["relu1_test.cc"], + tags = [ + "no_oss", + "tflite_not_portable_ios", + ], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + "@flatbuffers", + ], +) + +tf_cc_test( name = "activations_test", size = "small", srcs = ["activations_test.cc"], |