aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/BUILD
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-05-16 03:43:10 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-16 03:45:49 -0700
commit07bb8c1bbc93fe1162d247511c89c136273ddd07 (patch)
treeefd25e3aa8244e565bad28686053799bb5a8f3ba /tensorflow/contrib/lite/kernels/BUILD
parent8cfbc0c54e71b221152af938d4c8c5b2dea0f07d (diff)
Implementation of transpose_conv
PiperOrigin-RevId: 196806646
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r--tensorflow/contrib/lite/kernels/BUILD14
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD
index 6e2e790517..b7291dd379 100644
--- a/tensorflow/contrib/lite/kernels/BUILD
+++ b/tensorflow/contrib/lite/kernels/BUILD
@@ -177,6 +177,7 @@ cc_library(
"svdf.cc",
"topk_v2.cc",
"transpose.cc",
+ "transpose_conv.cc",
"unidirectional_sequence_lstm.cc",
"unidirectional_sequence_rnn.cc",
],
@@ -920,6 +921,19 @@ tf_cc_test(
],
)
+tf_cc_test(
+ name = "transpose_conv_test",
+ size = "small",
+ srcs = ["transpose_conv_test.cc"],
+ tags = ["tflite_not_portable_ios"],
+ deps = [
+ ":builtin_ops",
+ "//tensorflow/contrib/lite:framework",
+ "//tensorflow/contrib/lite/kernels:test_util",
+ "@com_google_googletest//:gtest",
+ ],
+)
+
filegroup(
name = "all_files",
srcs = glob(