diff options
author | 2018-05-16 03:43:10 -0700 | |
---|---|---|
committer | 2018-05-16 03:45:49 -0700 | |
commit | 07bb8c1bbc93fe1162d247511c89c136273ddd07 (patch) | |
tree | efd25e3aa8244e565bad28686053799bb5a8f3ba /tensorflow/contrib/lite/kernels/BUILD | |
parent | 8cfbc0c54e71b221152af938d4c8c5b2dea0f07d (diff) |
Implementation of transpose_conv
PiperOrigin-RevId: 196806646
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 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( |