code: def create_tensor_split_graph(
      sess: Session,
      input_value: Any,
      input_dtype: Any,
      num_outputs: int,
      num_splits: List[int],
      paddings: Optional[List[int]] = None,
  ) -> List[Tensor]:
      del sess
  
      const_input_op = constant_op.constant(
          input_value, dtype=input_dtype
      )
      return gen_tpu_ops.xla_split_nd(
          const_input_op, num_outputs, num_splits, paddings=paddings
      )
  
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 1 for calling a global function with example usage: create_tensor_split_graph(sess, input_value, input_dtype, num_outputs, num_splits, paddings) and returns: gen_tpu_ops