util_test.py 文件源码

def test_control_outputs(self):
    """Test for the ge.util.ControlOutputs class."""
    g0 = ops.Graph()
    with g0.as_default():
      a0 = constant_op.constant(1)
      b0 = constant_op.constant(2)
      x0 = constant_op.constant(3)
      with ops.control_dependencies([x0.op]):
        c0 = math_ops.add(a0, b0)  # pylint: disable=unused-variable
    control_outputs = ge.util.ControlOutputs(g0).get_all()
    self.assertEqual(len(control_outputs), 1)
    self.assertEqual(len(control_outputs[x0.op]), 1)
    self.assertIs(list(control_outputs[x0.op])[0], c0.op)