code: def testIsMegamorphic(self):
      @function.Defun(compiled=True)
      def CompiledFunction(x):
          return math_ops.log(x)
  
      with session_lib.Session(config=NoRewriteSessionConfig()) as sess:
          x = array_ops.placeholder(dtypes.float32)
          y = CompiledFunction(x)
  
          # Make the cluster go megamorphic by running it with lots of shape
          # signatures where the cluster is executed with each signature only a few
          # times.  Then check that we don't compile the cluster ever again.
  
          for shape in range(10, 50):
              for _ in range(0, 49):
                  sess.run(y, feed_dict={x: [0.0] * shape})
  
          for _ in range(0, 50):
              run_metadata = config_pb2.RunMetadata()
              sess.run(
                  y,
                  feed_dict={x: [0.0] * 60},
                  run_metadata=run_metadata,
                  options=config_pb2.RunOptions(
                      trace_level=config_pb2.RunOptions.FULL_TRACE
                  ),
              )
              self.assertTrue(
                  InLabels(
                      RunMetadataLabels(run_metadata), "_XlaCompile"
                  )
              )
              self.assertFalse(
                  InLabels(RunMetadataLabels(run_metadata), "_XlaRun")
              )
  
imports: import os
  import numpy
  from tensorflow.compiler.tests import test_utils
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python.client import session
  from tensorflow.python.compiler.xla import jit
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import cond
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the LazyCompilationTest obj's underlying member method code with example usage: obj.testIsMegamorphic() and returns: math_ops