diff options
author | Justine Tunney <jart@google.com> | 2017-05-09 19:22:09 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-10 17:18:03 -0700 |
commit | 236ef2c9c7b77f851609cb4ee2d82c1428b66c8c (patch) | |
tree | a3e859e8aea1506148014f23236ce0aaa4adb920 /tensorflow/contrib/tfprof | |
parent | b808475cc65351f16d383a53547f82f9632aeedd (diff) |
Make pprof definition deterministic
PiperOrigin-RevId: 155584393
Diffstat (limited to 'tensorflow/contrib/tfprof')
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 |