Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow.
No code changes, only doc-strings and comments. PiperOrigin-RevId: 243819553
This commit is contained in:
parent
e350128d64
commit
1d92ee8e3d
@ -947,7 +947,8 @@ class DebugDumpDir(object):
|
||||
3) The parallel_iteration attribute of while-loop Enter ops are set to 1.
|
||||
|
||||
Returns:
|
||||
A dict mapping device names (`str`s) to reconstructed `tf.GraphDef`s.
|
||||
A dict mapping device names (`str`s) to reconstructed
|
||||
`tf.compat.v1.GraphDef`s.
|
||||
"""
|
||||
non_debug_graphs = {}
|
||||
for key in self._debug_graphs:
|
||||
|
@ -132,7 +132,7 @@ class GradientsDebugger(object):
|
||||
|
||||
# Create a train op under the grad_debugger context.
|
||||
with grad_debugger:
|
||||
train_op = tf.train.GradientDescentOptimizer(z)
|
||||
train_op = tf.compat.v1.train.GradientDescentOptimizer(z)
|
||||
|
||||
# Now we can reflect through grad_debugger to get the gradient tensor
|
||||
# with respect to y.
|
||||
@ -195,7 +195,7 @@ class GradientsDebugger(object):
|
||||
# Create a train op under the grad_debugger context.
|
||||
grad_debugger = tf_debug.GradientsDebugger()
|
||||
with grad_debugger.watch_gradients_by_tensors(y):
|
||||
train_op = tf.train.GradientDescentOptimizer(z)
|
||||
train_op = tf.compat.v1.train.GradientDescentOptimizer(z)
|
||||
|
||||
# Now we can reflect through grad_debugger to get the gradient tensor
|
||||
# with respect to y.
|
||||
@ -247,7 +247,7 @@ class GradientsDebugger(object):
|
||||
# Create a train op under the grad_debugger context.
|
||||
grad_debugger = tf_debug.GradientsDebugger()
|
||||
with grad_debugger.watch_gradients_by_tensor_names(r"(x|y):0$"):
|
||||
train_op = tf.train.GradientDescentOptimizer(z)
|
||||
train_op = tf.compat.v1.train.GradientDescentOptimizer(z)
|
||||
|
||||
# Now we can reflect through grad_debugger to get the gradient tensor
|
||||
# with respect to x and y.
|
||||
|
@ -482,8 +482,8 @@ class DebugGraph(object):
|
||||
def reconstruct_non_debug_graph_def(debug_graph_def):
|
||||
"""Reconstruct original (non-debugger-decorated) partition GraphDef.
|
||||
|
||||
This method strips the input `tf.GraphDef` of the Copy* and Debug*-type nodes
|
||||
inserted by the debugger.
|
||||
This method strips the input `tf.compat.v1.GraphDef` of the Copy* and
|
||||
Debug*-type nodes inserted by the debugger.
|
||||
|
||||
The reconstructed partition graph is identical to the original (i.e.,
|
||||
non-debugger-decorated) partition graph except in the following respects:
|
||||
@ -494,10 +494,11 @@ def reconstruct_non_debug_graph_def(debug_graph_def):
|
||||
3) The parallel_iteration attribute of while-loop Enter ops are set to 1.
|
||||
|
||||
Args:
|
||||
debug_graph_def: The debugger-decorated `tf.GraphDef`, with the
|
||||
debug_graph_def: The debugger-decorated `tf.compat.v1.GraphDef`, with the
|
||||
debugger-inserted Copy* and Debug* nodes.
|
||||
|
||||
Returns:
|
||||
The reconstructed `tf.GraphDef` stripped of the debugger-inserted nodes.
|
||||
The reconstructed `tf.compat.v1.GraphDef` stripped of the debugger-inserted
|
||||
nodes.
|
||||
"""
|
||||
return DebugGraph(debug_graph_def).non_debug_graph_def
|
||||
|
@ -12,7 +12,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
"""Tests for debugger functionalities in tf.Session with grpc:// URLs.
|
||||
"""Tests for debugger functionalities in tf.compat.v1.Session with grpc:// URLs.
|
||||
|
||||
This test focus on grpc:// debugging of distributed (gRPC) sessions.
|
||||
"""
|
||||
|
@ -12,7 +12,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
"""Tests for debugger functionalities in tf.Session with grpc:// URLs.
|
||||
"""Tests for debugger functionalities in tf.compat.v1.Session with grpc:// URLs.
|
||||
|
||||
This test file focuses on the grpc:// debugging of local (non-distributed)
|
||||
tf.Sessions.
|
||||
|
@ -134,7 +134,7 @@ class SessionDebugTestBase(test_util.TensorFlowTestCase):
|
||||
"""Run fetches with debugging and obtain DebugDumpDir.
|
||||
|
||||
Args:
|
||||
sess: the tf.Session to be used.
|
||||
sess: the tf.compat.v1.Session to be used.
|
||||
fetches: fetches of the Session.run().
|
||||
feed_dict: feed dict for the Session.run().
|
||||
debug_ops: name(s) of the debug ops to be used.
|
||||
|
@ -128,8 +128,8 @@ def _send_call_tracebacks(destinations,
|
||||
call_key: The key of the execution call, as a string. For graph execution,
|
||||
this is a string describing the feeds, fetches (and targets) names of the
|
||||
`tf.Session.run` call. For eager execution, this is ignored.
|
||||
graph: A Python `tf.Graph` object (i.e., *not* a `tf.GraphDef`), which
|
||||
contains op tracebacks, if applicable.
|
||||
graph: A Python `tf.Graph` object (i.e., *not* a `tf.compat.v1.GraphDef`),
|
||||
which contains op tracebacks, if applicable.
|
||||
send_source: Whether the source files involved in the op tracebacks but
|
||||
outside the TensorFlow library are to be sent.
|
||||
"""
|
||||
@ -199,8 +199,8 @@ def send_graph_tracebacks(destinations,
|
||||
run_key: A string describing the feeds, fetches (and targets) names of the
|
||||
`tf.Session.run` call.
|
||||
origin_stack: The traceback of the `tf.Session.run()` invocation.
|
||||
graph: A Python `tf.Graph` object (i.e., *not* a `tf.GraphDef`), which
|
||||
contains op tracebacks.
|
||||
graph: A Python `tf.Graph` object (i.e., *not* a `tf.compat.v1.GraphDef`),
|
||||
which contains op tracebacks.
|
||||
send_source: Whether the source files involved in the op tracebacks but
|
||||
outside the TensorFlow library are to be sent.
|
||||
"""
|
||||
|
@ -97,7 +97,7 @@ class NodeStepper(object):
|
||||
c = tf.add(a, b, name="c")
|
||||
d = tf.multiply(a, c, name="d")
|
||||
|
||||
sess = tf.Session()
|
||||
sess = tf.compat.v1.Session()
|
||||
sess.run(tf.initialize_all_varialbes())
|
||||
stepper = NodeStepper(sess, d)
|
||||
|
||||
|
@ -180,7 +180,7 @@ class TensorBoardDebugWrapperSession(GrpcDebugWrapperSession):
|
||||
"""Constructor of TensorBoardDebugWrapperSession.
|
||||
|
||||
Args:
|
||||
sess: The `tf.Session` instance to be wrapped.
|
||||
sess: The `tf.compat.v1.Session` instance to be wrapped.
|
||||
grpc_debug_server_addresses: gRPC address(es) of debug server(s), as a
|
||||
`str` or a `list` of `str`s. E.g., "localhost:2333",
|
||||
"grpc://localhost:2333", ["192.168.0.7:2333", "192.168.0.8:2333"].
|
||||
|
@ -31,7 +31,7 @@ from tensorflow.python.training import session_run_hook
|
||||
class LocalCLIDebugHook(session_run_hook.SessionRunHook):
|
||||
"""Command-line-interface debugger hook.
|
||||
|
||||
Can be used as a hook for `tf.train.MonitoredSession`s and
|
||||
Can be used as a hook for `tf.compat.v1.train.MonitoredSession`s and
|
||||
`tf.estimator.Estimator`s. Provides a substitute for
|
||||
`tfdbg.LocalCLIDebugWrapperSession` in cases where the session is not directly
|
||||
available.
|
||||
@ -156,7 +156,7 @@ class LocalCLIDebugHook(session_run_hook.SessionRunHook):
|
||||
class DumpingDebugHook(session_run_hook.SessionRunHook):
|
||||
"""A debugger hook that dumps debug data to filesystem.
|
||||
|
||||
Can be used as a hook for `tf.train.MonitoredSession`s and
|
||||
Can be used as a hook for `tf.compat.v1.train.MonitoredSession`s and
|
||||
`tf.estimator.Estimator`s.
|
||||
"""
|
||||
|
||||
@ -235,7 +235,7 @@ class GrpcDebugHook(session_run_hook.SessionRunHook):
|
||||
When the arguments of debug_utils.watch_graph changes, strongly consider
|
||||
changing arguments here too so that features are available to tflearn users.
|
||||
|
||||
Can be used as a hook for `tf.train.MonitoredSession`s and
|
||||
Can be used as a hook for `tf.compat.v1.train.MonitoredSession`s and
|
||||
`tf.estimator.Estimator`s.
|
||||
"""
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user