diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-09-06 17:57:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-06 18:01:27 -0700 |
commit | e722358e7e96dd2aa20d7e2c56336e76845daa6a (patch) | |
tree | a74960670ce4bacad0909fc913097bcc3e27ed18 /tensorflow/python/profiler | |
parent | f8a43f9d63ce90f10852d69e40fbb9fe849fc190 (diff) |
Merge changes from github.
END_PUBLIC
---
Commit 607816029 authored by Eugene Brevdo<ebrevdo@google.com>
Committed by TensorFlower Gardener<gardener@tensorflow.org>:
Extended ScratchSpace to expose its underlying scratch tensor object.
PiperOrigin-RevId: 167649551
---
Commit db43fe68e authored by A. Unique TensorFlower<gardener@tensorflow.org>
Committed by TensorFlower Gardener<gardener@tensorflow.org>:
Add fast math attributes to all generated methods when fast math enabled.
RELNOTES: n/a
PiperOrigin-RevId: 167646637
---
Commit aebe8cc6f authored by A. Unique TensorFlower<gardener@tensorflow.org>
Committed by TensorFlower Gardener<gardener@tensorflow.org>:
Call HloComputation.Accept instead of HloInstruction.Accept to get all instructions profiled.
RELNOTES: n/a
PiperOrigin-RevId: 167640259
---
Commit 0ab137cd8 authored by A. Unique TensorFlower<gardener@tensorflow.org>
Committed by TensorFlower Gardener<gardener@tensorflow.org>:
BEGIN_PUBLIC
Automated g4 rollback of changelist 167604306
PiperOrigin-RevId: 167800256
Diffstat (limited to 'tensorflow/python/profiler')
-rw-r--r-- | tensorflow/python/profiler/model_analyzer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/profiler/model_analyzer.py b/tensorflow/python/profiler/model_analyzer.py index 5345949664..a1fe47982f 100644 --- a/tensorflow/python/profiler/model_analyzer.py +++ b/tensorflow/python/profiler/model_analyzer.py @@ -117,7 +117,7 @@ class Profiler(object): ```python Typical use case: # Currently we are only allowed to create 1 profiler per process. - profiler = Profile(sess.graph) + profiler = Profiler(sess.graph) for i in xrange(total_steps): if i % 10000 == 0: @@ -174,7 +174,7 @@ class Profiler(object): """Add statistics of a step. Args: - step: A step uint64 used to identify the RunMetadata. Must be different + step: int, A step used to identify the RunMetadata. Must be different across different AddStep() calls. run_meta: RunMetadata proto that contains statistics of a session run. """ |