class_docstr: Test the hint to stub functionality.
class_docstr_tok: ["Test","the","hint","to","stub","functionality","."]
docstr: This tests a scaled add which has 3 inputs and 2 outputs.
code: def testScaleAndBiasAndIdentity(self):
      """This tests a scaled add which has 3 inputs and 2 outputs."""
      with ops.Graph().as_default():
          a = array_ops.constant(1.0)
          x = array_ops.constant([2.0, 3.0])
          b = array_ops.constant([4.0, 5.0])
  
          def _scaled_and_bias_and_identity(a, x, b):
              custom = op_hint.OpHint("scale_and_bias_and_identity")
              a, x, b = custom.add_inputs(a, x, b)
              return custom.add_outputs(a * x + b, x)
  
          output = array_ops.identity(
              _scaled_and_bias_and_identity(a, x, b), name="ModelOutput"
          )
  
          with self.cached_session() as sess:
              # make sure one identity for each input (3) and output (2) => 3 + 2 = 5
              # +1 for the final output
              self.assertEqual(
                  self._countIdentities(sess.graph_def.node), 6
              )
  
              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(
                      [
                          "scale_and_bias_and_identity",
                          "Const",
                          "Identity",
                          "Pack",
                      ]
                  ),
              )
  
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 4 for calling the ConvertTestOpHint obj's underlying member method code with example usage: obj.testScaleAndBiasAndIdentity() and returns: custom