diff options
author | Jared Duke <jdduke@google.com> | 2018-06-20 11:48:15 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-20 11:51:26 -0700 |
commit | 4efefb90391b12c95339ed3b46a02b62ea5e195d (patch) | |
tree | bb3f9bb986b89287983ea8e7c35827993aad7206 /tensorflow/contrib/lite/kernels/BUILD | |
parent | e51df5918020cdfada26022240091e5529f7da60 (diff) |
Implement TFLite Shape operator
PiperOrigin-RevId: 201389618
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 bb5558443b..a77897a173 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -168,6 +168,7 @@ cc_library( "reshape.cc", "resize_bilinear.cc", "select.cc", + "shape.cc", "skip_gram.cc", "slice.cc", "space_to_batch_nd.cc", @@ -994,6 +995,20 @@ tf_cc_test( ], ) +tf_cc_test( + name = "shape_test", + size = "small", + srcs = ["shape_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", + ], +) + filegroup( name = "all_files", srcs = glob( |