class_docstr: Test the hint to stub functionality.
class_docstr_tok: ["Test","the","hint","to","stub","functionality","."]
docstr: Makes a custom op swish and makes sure it gets converted as a unit.
code: def testSwishLiteHint(self):
      """Makes a custom op swish and makes sure it gets converted as a unit."""
      with ops.Graph().as_default():
          image = array_ops.constant([1.0, 2.0, 3.0, 4.0])
          swish_scale = array_ops.constant(1.0)
  
          def _swish(input_tensor, scale):
              custom = op_hint.OpHint("cool_activation")
              input_tensor, scale = custom.add_inputs(
                  input_tensor, scale
              )
              output = (
                  math_ops.sigmoid(input_tensor) * input_tensor * scale
              )
              (output,) = custom.add_outputs(output)
              return output
  
          output = array_ops.identity(
              _swish(image, swish_scale), name="ModelOutput"
          )
  
          with self.cached_session() as sess:
              # check if identities have been put into the graph (2 input, 1 output,
              # and 1 final output).
              self.assertEqual(
                  self._countIdentities(sess.graph_def.node), 4
              )
  
              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(["cool_activation", "Const", "Identity"]),
              )
  
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 3 for calling the ConvertTestOpHint obj's underlying member method code with example usage: obj.testSwishLiteHint() and returns: output