diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-06-27 16:33:00 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-27 16:37:09 -0700 |
| commit | 50b999a8336d19400ab75aea66fe46eca2f5fe0b (patch) | |
| tree | 7cba4f4af6b131c253b65ff9f2923e851184668c /tensorflow/contrib/tfprof | |
| parent | d6d58a3a1785785679af56c0f8f131e7312b8226 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 160344052
Diffstat (limited to 'tensorflow/contrib/tfprof')
| -rw-r--r-- | tensorflow/contrib/tfprof/README.md | 23 | ||||
| -rw-r--r-- | tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py | 2 |
2 files changed, 24 insertions, 1 deletions
diff --git a/tensorflow/contrib/tfprof/README.md b/tensorflow/contrib/tfprof/README.md index 824ba4c09b..4fa1ccea69 100644 --- a/tensorflow/contrib/tfprof/README.md +++ b/tensorflow/contrib/tfprof/README.md @@ -1,3 +1,26 @@ # tfprof: TensorFlow Profiler and Beyond # Full Document in tensorflow/tools/tfprof/README.md + +Author: Xin Pan (xpan@google.com, github: panyx0718), Jon Shlens, Yao Zhang + +Consultants: Jon Shlens, Pete Warden + +###Major Features + +1. Measure model parameters, float operations, tensor shapes. +2. Profile op execution times, requested memory size and device placement. +3. Inspect checkpoint tensors' shapes and their values. +4. Selectively group, filter, account and order ops. + +####tfprof supports 3 views to organize TensorFlow model profiles + + * code view: Stats are associated your Python codes and organized as call stacks. + * scope view: Stats are organized as name scope hierarchies. + * graph view: Stats are organized as Tensorflow Op graph. + +####For each view, there are 3 ways to display outputs: + + * stdout: Results are written to stdout. + * timeline: Visualized in chrome browser as time series. + * file: Results are dumped to file. diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py index 9c59df3117..71468dde37 100644 --- a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py +++ b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py @@ -89,7 +89,7 @@ def _run_loop_model(): class RunMetadataTest(test.TestCase): def testGPU(self): - if not test.is_gpu_available(): + if not test.is_gpu_available(cuda_only=True): return ops.reset_default_graph() |