def METHOD_NAME(self):
    class OnlyOverheadLayer(tf.keras.layers.Layer):
        def call(self, x):
            return x
    layer = OnlyOverheadLayer()
    x = tf.convert_to_tensor([[1.0]])
    def fn():
        layer(x)
    self._run(fn, 10000)