if func.signature.name.startswith('cond_false'):
        self.assertTrue(
            any(map(_is_quantized_function_call_node, func.node_def))
        )
      elif func.signature.name.startswith('cond_true'):
        self.assertFalse(
            any(map(_is_quantized_function_call_node, func.node_def))
        )

  # Run this test only with the eager mode.
  @test_util.run_v2_only