docstr: Multiple TensorFlow function calls inside StableHLO.
docstr_tok: ["Multiple","TensorFlow","function","calls","inside","StableHLO","."]
code: def test_tf_call_function_multiple_funcs(self):
      """Multiple TensorFlow function calls inside StableHLO."""
      x = np.int32(2)
      y = np.int32(3)
      res = (x + y) + (x + y)
  
      @function.Defun(dtypes.int32, dtypes.int32)
      def foo(x, y):
          return x + y
  
      @function.Defun(dtypes.int32, dtypes.int32)
      def bar(x, y):
          return foo(x, y)
  
      def f(x, y):
          module, version = serialize(
              """
  module @jit_fun_flat_jax {
    func.func public @main(%arg0: tensor<i32>, %arg1: tensor<i32>) -> tensor<i32> {
      %0 = stablehlo.custom_call @tf.call_tf_function(%arg0, %arg1) {
        tf.backend_config = {called_index = 0}
      } : (tensor<i32>, tensor<i32>) -> tensor<i32>
      %1 = stablehlo.custom_call @tf.call_tf_function(%arg0, %arg1) {
        tf.backend_config = {called_index = 1}
      } : (tensor<i32>, tensor<i32>) -> tensor<i32>
      %2 = stablehlo.custom_call @tf.call_tf_function(%0, %1) {
        tf.backend_config = {called_index = 1}
      } : (tensor<i32>, tensor<i32>) -> tensor<i32>
      return %2 : tensor<i32>
    }
  }
  """
          )
          return xla.call_module(
              [x, y],
              version=version,
              module=module,
              Tout=[res.dtype],
              Sout=[res.shape],
              platforms=[self.testing_platform()],
              function_list=(foo, bar),
          )
  
      self._assertOpOutputMatchesExpected(f, (x, y), (res,))
  
imports: import os
  import re
  from typing import Optional, Sequence
  import unittest
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.mlir.stablehlo import stablehlo
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.ops import gen_xla_ops
  from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  from tensorflow.python.platform import googletest
  from tensorflow.python.platform import test
label_desc: Use image node_id 43 for calling the XlaCallModuleOpTest obj's underlying member method code with example usage: obj.test_tf_call_function_multiple_funcs() and returns: unknown, foo, xla