docstr: Test operations with static shapes.
docstr_tok: ["Test","operations","with","static","shapes","."]
code: def test_static_tf_op(self):
      """Test operations with static shapes."""
  
      @def_function.function(jit_compile=True)
      def compiled_f(x):
          return test_ops_for_light_outside_compilation.test_static_tf(
              x
          )
  
      with context.device("/gpu:0"):
          z = random_ops.random_normal([2, 2])
  
          self.assertFilecheck(
              compiled_f.experimental_get_compiler_ir(z)("hlo"),
              r"""
            CHECK: f32[2,2]{1,0} custom-call(f32[2,2]{1,0} [[v:.*]]), custom_call_target="GenericTfCallbackGPU"
            CHECK: TestStaticTf
            """,
          )
  
          self.assertAllClose(compiled_f(z), z)
  
imports: from tensorflow.compiler.mlir.python.mlir_wrapper import filecheck_wrapper
  from tensorflow.compiler.tf2xla import test_ops_for_light_outside_compilation
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.platform import googletest
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the LightOutsideCompilationTest obj's underlying member method code with example usage: obj.test_static_tf_op() and returns: test_ops_for_light_outside_compilation