From 63a49c712edd3b2ee990a9f98b766b24190d3ccb Mon Sep 17 00:00:00 2001 From: Nupur Garg Date: Tue, 14 Aug 2018 18:28:47 -0700 Subject: Adds support for Eager delegate to tflite_diff. PiperOrigin-RevId: 208752057 --- tensorflow/contrib/lite/testing/BUILD | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'tensorflow/contrib/lite/testing/BUILD') diff --git a/tensorflow/contrib/lite/testing/BUILD b/tensorflow/contrib/lite/testing/BUILD index a788d41ba7..89912fd116 100644 --- a/tensorflow/contrib/lite/testing/BUILD +++ b/tensorflow/contrib/lite/testing/BUILD @@ -162,11 +162,12 @@ cc_library( ":test_runner", "//tensorflow/contrib/lite:builtin_op_data", "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/delegates/eager:delegate", "//tensorflow/contrib/lite/kernels:builtin_ops", ], ) -cc_test( +tf_cc_test( name = "tflite_driver_test", size = "small", srcs = ["tflite_driver_test.cc"], -- cgit v1.2.3