From 435599f5d896d7e1f721ffe6fd092d39efe2b027 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 9 Jun 2017 14:45:22 -0700 Subject: Adjust test sizes PiperOrigin-RevId: 158565259 --- tensorflow/contrib/tfprof/python/tools/tfprof/BUILD | 3 +++ 1 file changed, 3 insertions(+) (limited to 'tensorflow/contrib/tfprof') diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD b/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD index 93d0cb4099..d3b44804d6 100644 --- a/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD +++ b/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD @@ -84,6 +84,7 @@ py_library( tf_py_test( name = "tfprof_logger_test", + size = "small", srcs = ["tfprof_logger_test.py"], additional_deps = [ ":tfprof_logger", @@ -114,6 +115,7 @@ tf_py_wrap_cc( py_test( name = "print_model_analysis_test", + size = "small", srcs = ["print_model_analysis_test.py"], srcs_version = "PY2AND3", deps = [ @@ -140,6 +142,7 @@ py_library( py_test( name = "pprof_profiler_test", + size = "small", srcs = ["pprof_profiler_test.py"], main = "pprof_profiler_test.py", srcs_version = "PY2AND3", -- cgit v1.2.3