diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-10 10:22:57 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 10:29:57 -0700 |
commit | d2fa11acfd6b8a2a2663dc70cd899433bdde23e1 (patch) | |
tree | 55d9865cef5c19fec51556aa1b8b5afc66702efa /tensorflow/contrib/lite/kernels/BUILD | |
parent | a6df304e5ec4b0ad28884025558ecda414b9df07 (diff) |
Adding FakeQuant op support.
PiperOrigin-RevId: 203969441
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 e749edc5ee..edce73989c 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -155,6 +155,7 @@ cc_library( "embedding_lookup_sparse.cc", "exp.cc", "expand_dims.cc", + "fake_quant.cc", "floor.cc", "fully_connected.cc", "gather.cc", @@ -564,6 +565,19 @@ tf_cc_test( ) tf_cc_test( + name = "fake_quant_test", + size = "small", + srcs = ["fake_quant_test.cc"], + tags = ["tflite_not_portable_ios"], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "maximum_minimum_test", size = "small", srcs = ["maximum_minimum_test.cc"], |