code: def find_chunk_all_input_nodes(nodes: List[Node]) -> List:
      """
      Find non-compute input and output node names.
      input nodes are nodes used in the list
      output nodes are nodes will use nodes in the list
      """
      input_nodes = []
      for node in nodes:
          for input_node in node._input_nodes.keys():
              if (
                  input_node not in nodes
                  and input_node not in input_nodes
              ):
                  input_nodes.append(input_node)
      return input_nodes
  
imports: from typing import Any, Dict, List, Union
  from torch.fx.node import Node
  from colossalai.logging import get_dist_logger
label_desc: Use image node_id 10 for calling a global function with example usage: find_chunk_all_input_nodes(nodes) and returns: input_nodes