diff options
author | 2018-06-01 12:53:54 -0700 | |
---|---|---|
committer | 2018-06-01 12:57:00 -0700 | |
commit | b812f37e26889bb168fa0279a536b907c3fb5fdd (patch) | |
tree | 0b5804e7749d0b83a44748ca848917ad1554ceae /tensorflow/contrib/lite/kernels/BUILD | |
parent | 10b2b3b44a6f93f4fd414e8ac450587ece2207ae (diff) |
TFLite: adding tile and expand_dims ops.
PiperOrigin-RevId: 198913026
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index 0af659b5ca..cf5d0b4ce9 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -147,6 +147,7 @@ cc_library( "embedding_lookup.cc", "embedding_lookup_sparse.cc", "exp.cc", + "expand_dims.cc", "floor.cc", "fully_connected.cc", "gather.cc", @@ -176,6 +177,7 @@ cc_library( "strided_slice.cc", "sub.cc", "svdf.cc", + "tile.cc", "topk_v2.cc", "transpose.cc", "transpose_conv.cc", @@ -859,6 +861,20 @@ tf_cc_test( ) tf_cc_test( + name = "tile_test", + size = "small", + srcs = ["tile_test.cc"], + tags = ["tflite_not_portable_ios"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:builtin_op_data", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "comparisons_test", size = "small", srcs = [ @@ -936,12 +952,27 @@ tf_cc_test( ) tf_cc_test( + name = "expand_dims_test", + size = "small", + srcs = ["expand_dims_test.cc"], + tags = ["tflite_not_portable_ios"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:builtin_op_data", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "sparse_to_dense_test", size = "small", srcs = ["sparse_to_dense_test.cc"], tags = ["tflite_not_portable_ios"], deps = [ ":builtin_ops", + "//tensorflow/contrib/lite:builtin_op_data", "//tensorflow/contrib/lite:framework", "//tensorflow/contrib/lite/kernels:test_util", "@com_google_googletest//:gtest", |