diff options
| author |  Alan Chiao <alanchiao@google.com> | 2018-10-04 08:30:22 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-04 08:34:41 -0700 |
| commit | dcd7dd2d2e1ed7d8c26dd22dbbd2bac269c42e1e (patch) | |
| tree | 06798fad9258383b59ed80e1c30a751495ceb229 /tensorflow/contrib/lite/kernels/BUILD | |
| parent | 7b56d4ff7679ed59e3ea799054c5dcefd0600ab0 (diff) | |
Sparse output fully connected custom op.
PiperOrigin-RevId: 215741296
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 daaf6714cc..95e387814d 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -210,6 +210,7 @@ cc_library( "slice.cc", "space_to_batch_nd.cc", "space_to_depth.cc", + "sparse_output_fully_connected.cc", "sparse_to_dense.cc", "split.cc", "squeeze.cc", @@ -334,6 +335,23 @@ tf_cc_test( ) tf_cc_test( + name = "sparse_output_fully_connected_test", + size = "small", + srcs = ["sparse_output_fully_connected_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"], |