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/__init__.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/__init__.py')
-rw-r--r-- | tensorflow/python/__init__.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py index db2c5b1fd9..a2c86cbb6c 100644 --- a/tensorflow/python/__init__.py +++ b/tensorflow/python/__init__.py @@ -84,6 +84,7 @@ from tensorflow.python.ops import nn from tensorflow.python.ops import sets from tensorflow.python.ops import spectral_ops as spectral from tensorflow.python.ops.losses import losses +from tensorflow.python.profiler import profiler from tensorflow.python.user_ops import user_ops from tensorflow.python.util import compat from tensorflow.python.saved_model import saved_model @@ -235,6 +236,7 @@ _allowed_symbols.extend([ 'train', 'user_ops', 'layers', + 'profiler', ]) # Variables framework.versions: @@ -251,7 +253,7 @@ remove_undocumented(__name__, _allowed_symbols, [ control_flow_ops, confusion_matrix_m, functional_ops, histogram_ops, io_ops, losses, math_ops, metrics, nn, resource_loader, sets, script_ops, session_ops, sparse_ops, state_ops, string_ops, summary, tensor_array_ops, - train, layers + train, layers, profiler ]) # Special dunders that we choose to export: |