提交 0fd7e611 编写于 作者: S Shanqing Cai 提交者: TensorFlower Gardener

[tfdbg] Add reasons for run_v1_only decorators

PiperOrigin-RevId: 318320055
Change-Id: I26271c9339b0381dd624a8dcee8758083b1313a8
上级 b9b4645e
......@@ -70,7 +70,7 @@ def _assert_no_lines_match(pattern, lines):
"%s matched at least one line in %s." % (pattern, str(lines)))
@test_util.run_v1_only("b/120545219")
@test_util.run_v1_only("Requires tf.Session")
class ProfileAnalyzerListProfileTest(test_util.TensorFlowTestCase):
def testNodeInfoEmpty(self):
......@@ -322,7 +322,7 @@ class ProfileAnalyzerListProfileTest(test_util.TensorFlowTestCase):
_assert_at_least_one_line_matches(r"Device Total.*0\.009ms", prof_output)
@test_util.run_v1_only("b/120545219")
@test_util.run_v1_only("Requires tf.Session")
class ProfileAnalyzerPrintSourceTest(test_util.TensorFlowTestCase):
def setUp(self):
......
......@@ -31,6 +31,7 @@ from tensorflow.python.ops import variables
from tensorflow.python.platform import googletest
@test_util.run_v1_only("Requires tf.Session")
class DebugUtilsTest(test_util.TensorFlowTestCase):
@classmethod
......@@ -192,7 +193,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
self.assertEqual(["file:///tmp/tfdbg_1", "file:///tmp/tfdbg_2"],
watch_0.debug_urls)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_allNodes(self):
debug_utils.watch_graph(
self._run_options,
......@@ -227,7 +227,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
# Assert that the wildcard node name has been created.
self.assertIn("*", node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_nodeNameWhitelist(self):
debug_utils.watch_graph(
self._run_options,
......@@ -242,7 +241,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
sorted(["a1_init", "a1", "a1/Assign", "a1/read", "p1"]),
sorted(node_names))
@test_util.run_v1_only("b/120545219")
def testWatchGraph_opTypeWhitelist(self):
debug_utils.watch_graph(
self._run_options,
......@@ -268,7 +266,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
["DebugIdentity"], ["file:///tmp/tfdbg_1"])
self.assertEqual(["p1"], node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_tensorDTypeWhitelist(self):
debug_utils.watch_graph(
self._run_options,
......@@ -281,7 +278,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
["DebugIdentity"], ["file:///tmp/tfdbg_1"])
self.assertItemsEqual(["a1", "a1/Assign", "b", "b/Assign"], node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_nodeNameAndTensorDTypeWhitelists(self):
debug_utils.watch_graph(
self._run_options,
......@@ -295,7 +291,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
["DebugIdentity"], ["file:///tmp/tfdbg_1"])
self.assertItemsEqual(["a1", "a1/Assign"], node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_nodeNameBlacklist(self):
debug_utils.watch_graph_with_blacklists(
self._run_options,
......@@ -310,7 +305,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
sorted(["b_init", "b", "b/Assign", "b/read", "c", "s"]),
sorted(node_names))
@test_util.run_v1_only("b/120545219")
def testWatchGraph_opTypeBlacklist(self):
debug_utils.watch_graph_with_blacklists(
self._run_options,
......@@ -323,7 +317,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
["DebugIdentity"], ["file:///tmp/tfdbg_1"])
self.assertEqual(sorted(["p1", "s"]), sorted(node_names))
@test_util.run_v1_only("b/120545219")
def testWatchGraph_nodeNameAndOpTypeBlacklists(self):
debug_utils.watch_graph_with_blacklists(
self._run_options,
......@@ -337,7 +330,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
["DebugIdentity"], ["file:///tmp/tfdbg_1"])
self.assertEqual(["s"], node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_tensorDTypeBlacklists(self):
debug_utils.watch_graph_with_blacklists(
self._run_options,
......@@ -354,7 +346,6 @@ class DebugUtilsTest(test_util.TensorFlowTestCase):
self.assertNotIn("b/Assign", node_names)
self.assertIn("s", node_names)
@test_util.run_v1_only("b/120545219")
def testWatchGraph_nodeNameAndTensorDTypeBlacklists(self):
debug_utils.watch_graph_with_blacklists(
self._run_options,
......
......@@ -91,7 +91,8 @@ class GrpcDebugServerTest(test_util.TensorFlowTestCase):
server.stop_server().wait()
@test_util.run_v1_only("b/120545219")
@test_util.run_v1_only(
"GrpcDebugWrapperSession and GrpcDebugHookare are for tf.Session only")
class SessionDebugGrpcTest(session_debug_testlib.SessionDebugTestBase):
@classmethod
......@@ -354,7 +355,7 @@ class SessionDebugConcurrentTest(
return urls
@test_util.run_v1_only("b/120545219")
@test_util.run_v1_only("GrpcDebugWrapperSession is for tf.Session only")
class SessionDebugGrpcGatingTest(test_util.TensorFlowTestCase):
"""Test server gating of debug ops."""
......@@ -732,7 +733,7 @@ class SessionDebugGrpcGatingTest(test_util.TensorFlowTestCase):
self.assertEqual("DebugNumericSummary", debug_watch.debug_op)
@test_util.run_v1_only("b/120545219")
@test_util.run_v1_only("GrpcDebugWrapperSession is for tf.Session only")
class DelayedDebugServerTest(test_util.TensorFlowTestCase):
def testDebuggedSessionRunWorksWithDelayedDebugServerStartup(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册