aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tfprof
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-09 14:45:22 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-09 14:48:59 -0700
commit435599f5d896d7e1f721ffe6fd092d39efe2b027 (patch)
tree609a81818fef9c899a628461654eba6683537e56 /tensorflow/contrib/tfprof
parent0058c1f134c7de0733ffed754fe3a3091dea60ca (diff)
Adjust test sizes
PiperOrigin-RevId: 158565259
Diffstat (limited to 'tensorflow/contrib/tfprof')
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/BUILD3
1 files changed, 3 insertions, 0 deletions
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",