diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-04 10:19:42 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-04 10:23:59 -0700 |
commit | af23ae65db2585f4a18d0bc5f21f15e94805aa4f (patch) | |
tree | a805f64d0a85fa29ff69d204634379b80cdbcbf1 /tensorflow/python/profiler/profiler.py | |
parent | 11ec8b7cfdec0fd498182d0ad8f550b4a8ddaf13 (diff) |
Migrating tfprof python API to tensorflow/python/profiler
Migrating tfprof c++ to tensorflow/core/profiler
API changes:
New tf.profiler namespace.
Within tf.profiler namespace:
tf.profiler.advise # One-shot advise function
tf.profiler.profile # One-shot profile function
tf.profiler.Profiler # Multi-step profile/advise class
tf.profiler.write_op_log # Write profile for offline analysis
PiperOrigin-RevId: 160901831
Diffstat (limited to 'tensorflow/python/profiler/profiler.py')
-rw-r--r-- | tensorflow/python/profiler/profiler.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tensorflow/python/profiler/profiler.py b/tensorflow/python/profiler/profiler.py new file mode 100644 index 0000000000..19b253ab2f --- /dev/null +++ b/tensorflow/python/profiler/profiler.py @@ -0,0 +1,32 @@ +# Copyright 2016 The TensorFlow Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================== +"""tfprof is a tool that profile various aspect of TensorFlow model. + +@@Profiler +@@profile +@@advise +@@write_op_log +""" +from __future__ import absolute_import +from __future__ import division +from __future__ import print_function + +# pylint: disable=unused-import,wildcard-import +from tensorflow.python.profiler.model_analyzer import * +from tensorflow.python.profiler.tfprof_logger import * +from tensorflow.python.util.all_util import remove_undocumented + + +remove_undocumented(__name__, []) |