Move data of quantization_mnist_test to testdata/mnist
diff --git a/tensorflow/python/compiler/tensorrt/BUILD b/tensorflow/python/compiler/tensorrt/BUILD
index 0e4c43f..a8bf5ae 100644
--- a/tensorflow/python/compiler/tensorrt/BUILD
+++ b/tensorflow/python/compiler/tensorrt/BUILD
@@ -169,9 +169,15 @@
"//tensorflow/python/estimator:estimator",
],
data = [
- "test/testdata/checkpoint",
- "test/testdata/model.ckpt-46900.data-00000-of-00001",
- "test/testdata/model.ckpt-46900.index",
+ "test/testdata/mnist/checkpoint",
+ "test/testdata/mnist/model.ckpt-46900.data-00000-of-00001",
+ "test/testdata/mnist/model.ckpt-46900.index",
+ "test/testdata/mnist/gen_tftrt_model.py",
+ "test/testdata/mnist/tftrt_2.0_saved_model/saved_model.pb",
+ "test/testdata/mnist/tftrt_2.0_saved_model/assets/trt-serialized-engine.TRTEngineOp_0",
+ "test/testdata/mnist/tftrt_2.0_saved_model/variables/variables.data-00000-of-00002",
+ "test/testdata/mnist/tftrt_2.0_saved_model/variables/variables.data-00001-of-00002",
+ "test/testdata/mnist/tftrt_2.0_saved_model/variables/variables.index",
],
tags = [
"no_cuda_on_cpu_tap",
diff --git a/tensorflow/python/compiler/tensorrt/test/quantization_mnist_test.py b/tensorflow/python/compiler/tensorrt/test/quantization_mnist_test.py
index f5be385..20287ee 100644
--- a/tensorflow/python/compiler/tensorrt/test/quantization_mnist_test.py
+++ b/tensorflow/python/compiler/tensorrt/test/quantization_mnist_test.py
@@ -265,7 +265,7 @@
def testEval(self):
if not is_tensorrt_enabled():
return
- model_dir = test.test_src_dir_path('python/compiler/tensorrt/test/testdata')
+ model_dir = test.test_src_dir_path('python/compiler/tensorrt/test/testdata/mnist')
accuracy_tf_native = self._Run(
is_training=False,
diff --git a/tensorflow/python/compiler/tensorrt/test/testdata/checkpoint b/tensorflow/python/compiler/tensorrt/test/testdata/checkpoint
deleted file mode 100644
index a603e1a..0000000
--- a/tensorflow/python/compiler/tensorrt/test/testdata/checkpoint
+++ /dev/null
@@ -1,3 +0,0 @@
-model_checkpoint_path: "model.ckpt-46900"
-all_model_checkpoint_paths: "model.ckpt-0"
-all_model_checkpoint_paths: "model.ckpt-46900"
diff --git a/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.data-00000-of-00001 b/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.data-00000-of-00001
deleted file mode 100644
index 88a998f..0000000
--- a/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.data-00000-of-00001
+++ /dev/null
Binary files differ
diff --git a/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.index b/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.index
deleted file mode 100644
index 5379765..0000000
--- a/tensorflow/python/compiler/tensorrt/test/testdata/model.ckpt-46900.index
+++ /dev/null
Binary files differ
