^bb0(%arg2, %arg3):
            %1 = "tf.ReadVariableOp"(%var)
            %2 = "tf.OpB"(%1) {is_stateless = true}
            %3 = "tf.OpC"(%arg2, %2) {is_stateless = true}
            %4 = "tf.OpD"(%arg3, %2) {is_stateless = true}
            "tf.Yield"(%3, %4)
          }) {is_stateless = true}
          return %results#0, %results#1
        }
      ```
      would be transformed to
      ```mlir

       }, {
       ^bb0(%arg2, %arg3):
  %1 = "tf.ReadVariableOp"(%var)
         %2 = "tf.OpB"(%1) {is_stateless = true}
         %3 = "tf.OpC"(%arg2, %2) {is_stateless = true}
         %4 = "tf.OpD"(%arg3, %2) {is_stateless = true}
         "tf.Yield"(%3, %4)
       }) {is_stateless = true}
       return %results#0, %results#1
     }
   ```
   would be transformed to
   ```mlir

        %2 = "tf.OpC"(%0, %1) {_xla_map_outside_compilation = "0", _xla_outside_compilation = "from_launch"} : (tensor<2x2xi64>, tensor<2x2xi64>) -> tensor<2x2xi64>
        "tf.OpD"(%2) : (tensor<2x2xi64>) -> ()
        tf_device.return