diff options
author | 2018-04-26 19:35:10 -0700 | |
---|---|---|
committer | 2018-04-26 19:38:26 -0700 | |
commit | e41e70ed9827b81a07c42f68def80f3f61b70375 (patch) | |
tree | 04fadd1f072d787c13f145ebf9c9f4a17008e9ab /tensorflow/contrib/lite/kernels/BUILD | |
parent | 84b3322931fd6fd73ce4ab250a1bd3cdd6e138f6 (diff) |
Implement floor operator
PiperOrigin-RevId: 194490433
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index 80cefe83b2..689f9bfa71 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -145,6 +145,7 @@ cc_library( "embedding_lookup.cc", "embedding_lookup_sparse.cc", "exp.cc", + "floor.cc", "fully_connected.cc", "gather.cc", "hashtable_lookup.cc", @@ -438,6 +439,19 @@ tf_cc_test( ) tf_cc_test( + name = "floor_test", + size = "small", + srcs = ["floor_test.cc"], + tags = ["tflite_not_portable_ios"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "unidirectional_sequence_lstm_test", size = "small", srcs = ["unidirectional_sequence_lstm_test.cc"], |