diff options
author | 2017-12-14 21:26:55 -0800 | |
---|---|---|
committer | 2017-12-14 21:31:06 -0800 | |
commit | dbcb1ffcca6a3c52e3c109a1739018350bc41925 (patch) | |
tree | aa4f440d1f8f4382cea7a1b4e4cfc623da2a76f0 /tensorflow/contrib/lite/kernels/BUILD | |
parent | f806269602219d5095265d036f294cc9a6260971 (diff) |
Support BatchToSpaceND in TFLite
The internal implementation only support 4D tensors for now.
The dimension has to be 1 batch + 2 spatial + 1 other.
The most common format within this restriction is NHWC.
Cropping is not supported by the internal implementation.
PiperOrigin-RevId: 179143332
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index 3908960c33..cc02cddb3d 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -77,6 +77,7 @@ cc_library( "activations.cc", "add.cc", "basic_rnn.cc", + "batch_to_space_nd.cc", "concatenation.cc", "conv.cc", "depthwise_conv.cc", @@ -157,6 +158,18 @@ tf_cc_test( ) tf_cc_test( + name = "batch_to_space_nd_test", + size = "small", + srcs = ["batch_to_space_nd_test.cc"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "concatenation_test", size = "small", srcs = ["concatenation_test.cc"], |