def testExtractGatedGrpcTensorsFoundNoGatedGrpcOps(self):
with tf.Session() as sess:
z, run_options = self._createTestGraphAndRunOptions(sess,
gated_grpc=False)
sess.run(tf.global_variables_initializer())
run_metadata = tf.RunMetadata()
self.assertAllClose(
[10.0], sess.run(z, options=run_options, run_metadata=run_metadata))
graph_wrapper = debug_graphs_helper.DebugGraphWrapper(
run_metadata.partition_graphs[0])
gated_debug_ops = graph_wrapper.get_gated_grpc_tensors()
self.assertEqual([], gated_debug_ops)
debug_graphs_helper_test.py 文件源码
python
阅读 27
收藏 0
点赞 0
评论 0
评论列表
文章目录