code: def create_tensor_roundtrip_graph(
      sess: Session,
      value: Any,
      dtype: Any,
      num_partitions: List[int],
      paddings: Optional[List[int]] = None,
  ) -> Tensor:
      del sess
  
      const_input_op = constant_op.constant(value, dtype=dtype)
      split = gen_tpu_ops.xla_split_nd(
          const_input_op,
          np.prod(num_partitions),
          num_partitions,
          paddings=paddings,
      )
      concat = gen_tpu_ops.xla_concat_nd(
          split, num_partitions, paddings
      )
      return math_ops.equal(const_input_op, concat)
  
imports: from typing import Any, List, Optional
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.client.session import Session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework.ops import control_dependencies
  from tensorflow.python.framework.ops import Tensor
  from tensorflow.python.ops import gen_tpu_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling a global function with example usage: create_tensor_roundtrip_graph(sess, value, dtype, num_partitions, paddings) and returns: math_ops