diff options
author | 2018-06-15 23:32:33 -0700 | |
---|---|---|
committer | 2018-06-15 23:35:03 -0700 | |
commit | 990e1f218c7180b2ebf407b8ec06d59936e9cc12 (patch) | |
tree | 0386a3f0afcf8468ddaf0d6e9616b6b3f1e82df5 /tensorflow/contrib/lite/kernels/BUILD | |
parent | df4ff7833725452c4ede1bf58b7523bafff3ecef (diff) |
TFLite Custom op for object detection postprocessing.
PiperOrigin-RevId: 200820561
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index cf5d0b4ce9..0b70c8ffa3 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -174,6 +174,7 @@ cc_library( "sparse_to_dense.cc", "split.cc", "squeeze.cc", + "ssd_postprocessing.cc", "strided_slice.cc", "sub.cc", "svdf.cc", @@ -247,6 +248,20 @@ tf_cc_test( ) tf_cc_test( + name = "ssd_postprocess_test", + size = "small", + srcs = ["ssd_postprocess_test.cc"], + tags = ["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"], |