code: def testIsNotMegamorphic(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)
  
          # Run the cluster with lots of shape signatures, but in a way that it
          # isn't megamorphic (i.e. each shape signature sees a lot of executions).
          # Then check that the cluster has not been marked as megamorphic.
  
          for shape in range(10, 50):
              for _ in range(0, 1000):
                  sess.run(y, feed_dict={x: [0.0] * shape})
  
          for _ in range(0, 10):
              sess.run(y, feed_dict={x: [0.0] * 60})
  
          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.assertTrue(
              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 3 for calling the LazyCompilationTest obj's underlying member method code with example usage: obj.testIsNotMegamorphic() and returns: math_ops