code: def remove_zeros(split_sections: List[int]):
      """
      Remove zeros from the list and get the index mapping dict from getitem
      in split node to getitem in new split node
      """
      new_split_sections, index_mapping = [], {}
      idx = 0
      for i in range(len(split_sections)):
          if split_sections[i] > 0:
              new_split_sections.append(split_sections[i])
              index_mapping[i] = idx
              idx += 1
  
      return new_split_sections, index_mapping
  
imports: import itertools
  import logging
  import operator
  from typing import Any, Callable, List, Optional, Sequence, Set, Tuple, Union
  from typing_extensions import TypeAlias
  import torch
  from torch._dynamo.utils import counters
  from ..pattern_matcher import Arg, CallFunction, CallFunctionVarArgs, CallMethodVarArgs, config_flag, FailedMatch, get_arg_value, Ignored, KeywordArg, ListOf, Match, MatchContext, MULTIPLE, PatternExpr, register_graph_pattern, RepeatedExpr
  from .pre_grad import merge_getitem_cat_pass, merge_splits_pass, normalization_pass, split_cat_pass, unbind_stack_pass
label_desc: Use image node_id 14 for calling a global function with example usage: remove_zeros(split_sections) and returns: new_split_sections,  index_mapping