class_docstr: Test the hint to stub functionality.
class_docstr_tok: ["Test","the","hint","to","stub","functionality","."]
docstr: Tests if two functions are converted correctly.
code: def testTwoFunctions(self):
      """Tests if two functions are converted correctly."""
      with ops.Graph().as_default():
          a = array_ops.constant([1.0])
          b = array_ops.constant([1.0])
  
          def _double_values(x):
              custom = op_hint.OpHint("add_test")
              (x,) = custom.add_inputs(x)
              output = math_ops.multiply(x, x)
              (output,) = custom.add_outputs(output)
              return output
  
          output = array_ops.identity(
              math_ops.add(_double_values(a), _double_values(b)),
              name="ModelOutput",
          )
  
          with self.cached_session() as sess:
              # make sure one identity for each input (2) and output (2) => 2 + 2
              # +1 for the final output
              self.assertEqual(
                  self._countIdentities(sess.graph_def.node), 5
              )
              stubbed_graphdef = op_hint.convert_op_hints_to_stubs(
                  graph_def=sess.graph_def
              )
              self.assertEqual(
                  self._getGraphOpTypes(
                      stubbed_graphdef,
                      output_nodes=[
                          op_hint._tensor_name_base(output.name)
                      ],
                  ),
                  set(["add_test", "Const", "Identity", "AddV2"]),
              )
  
imports: from unittest import mock
  import numpy
  from tensorflow.lite.python import convert
  from tensorflow.lite.python import op_hint
  from tensorflow.lite.python.interpreter import Interpreter
  from tensorflow.lite.python.metrics import converter_error_data_pb2
  from tensorflow.lite.python.metrics.wrapper import metrics_wrapper
  from tensorflow.lite.toco import toco_flags_pb2
  from tensorflow.python.client import session
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.framework.graph_util_impl import _bfs_for_reachable_nodes
  from tensorflow.python.framework.graph_util_impl import _extract_graph_summary
  from tensorflow.python.framework.graph_util_impl import _node_name
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the ConvertTestOpHint obj's underlying member method code with example usage: obj.testTwoFunctions() and returns: output