aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tfprof
diff options
context:
space:
mode:
authorGravatar Justine Tunney <jart@google.com>2017-05-09 19:22:09 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-05-10 17:18:03 -0700
commit236ef2c9c7b77f851609cb4ee2d82c1428b66c8c (patch)
treea3e859e8aea1506148014f23236ce0aaa4adb920 /tensorflow/contrib/tfprof
parentb808475cc65351f16d383a53547f82f9632aeedd (diff)
Make pprof definition deterministic
PiperOrigin-RevId: 155584393
Diffstat (limited to 'tensorflow/contrib/tfprof')
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/BUILD4
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler.py2
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler_test.py2
3 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD b/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD
index 0a920780c2..c96f6719e7 100644
--- a/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD
+++ b/tensorflow/contrib/tfprof/python/tools/tfprof/BUILD
@@ -119,7 +119,7 @@ py_library(
name = "pprof_profiler",
srcs = ["pprof_profiler.py"],
srcs_version = "PY2AND3",
- deps = ["@pprof_profile_proto//:pprof_proto_py"],
+ deps = ["@com_google_pprof//:pprof_proto_py"],
)
py_test(
@@ -134,7 +134,7 @@ py_test(
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform_test",
- "@pprof_profile_proto//:pprof_proto_py",
+ "@com_google_pprof//:pprof_proto_py",
],
)
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler.py b/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler.py
index c3fea915a3..c57e45748d 100644
--- a/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler.py
+++ b/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler.py
@@ -47,7 +47,7 @@ import string
import sys
import time
-from pprof import profile_pb2
+from proto import profile_pb2
if sys.version_info < (3,):
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler_test.py b/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler_test.py
index 13d3fb41ac..6487adf992 100644
--- a/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler_test.py
+++ b/tensorflow/contrib/tfprof/python/tools/tfprof/pprof_profiler_test.py
@@ -20,7 +20,7 @@ from __future__ import print_function
import gzip
-from pprof import profile_pb2
+from proto import profile_pb2
from tensorflow.contrib.tfprof.python.tools.tfprof import pprof_profiler
from tensorflow.core.framework import step_stats_pb2
from tensorflow.core.protobuf import config_pb2