TrainingExtensions/torch/src/python/aimet_torch/meta/connectedgraph.py
def _handle_ir_nodes_of_interest(self, ir_nodes_list: List[IrNode], passthrough_ops: List[str],
                                     input_ops_to_ignore: List[str]):
        """
        Update input and output connections of certain ir_nodes in ir_nodes_list (Tuple/ListConstructs, passthrough,
        input ops)
        :param ir_nodes_list: List of ir_nodes to update connections for
        :param passthrough_ops: List of op types to treat as passthrough
        :param input_ops_to_ignore: List of input op types to ignore
        """
        connections_to_ir_nodes_dict = self._create_connections_to_ir_nodes_dict(ir_nodes_list)
        for ir_node in ir_nodes_list:
            if ir_node.node_type in ['TupleConstruct', 'ListConstruct']:
                _handle_tuple_and_list_construct_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in ['TupleUnpack', 'ListUnpack']:
                _handle_tuple_unpack_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in passthrough_ops:
                _handle_passthrough_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in input_ops_to_ignore:
                _handle_input_ir_node_to_ignore(ir_node, connections_to_ir_nodes_dict)

        Update input and output connections of certain ir_nodes in ir_nodes_list (Tuple/ListConstructs, passthrough,
        input ops)
        :param ir_nodes_list: List of ir_nodes to update connections for
        :param passthrough_ops: List of op types to treat as passthrough
        :param input_ops_to_ignore: List of input op types to ignore
        
Update input and output connections of certain ir_nodes in ir_nodes_list (Tuple/ListConstructs, passthrough,
input ops)
Update input and output connections of certain ir_nodes in ir_nodes_list ( Tuple / ListConstructs passthrough input ops )
def _handle_ir_nodes_of_interest(self, ir_nodes_list: List[IrNode], passthrough_ops: List[str],
                                     input_ops_to_ignore: List[str]):
        connections_to_ir_nodes_dict = self._create_connections_to_ir_nodes_dict(ir_nodes_list)
        for ir_node in ir_nodes_list:
            if ir_node.node_type in ['TupleConstruct', 'ListConstruct']:
                _handle_tuple_and_list_construct_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in ['TupleUnpack', 'ListUnpack']:
                _handle_tuple_unpack_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in passthrough_ops:
                _handle_passthrough_ir_node(ir_node, connections_to_ir_nodes_dict)
            elif ir_node.node_type in input_ops_to_ignore:
                _handle_input_ir_node_to_ignore(ir_node, connections_to_ir_nodes_dict)
def _handle_ir_nodes_of_interest ( self , ir_nodes_list : List [ IrNode ] , passthrough_ops : List [ str ] , input_ops_to_ignore : List [ str ] ) : connections_to_ir_nodes_dict = self . _create_connections_to_ir_nodes_dict ( ir_nodes_list ) for ir_node in ir_nodes_list : if ir_node . node_type in [ 'TupleConstruct' , 'ListConstruct' ] : _handle_tuple_and_list_construct_ir_node ( ir_node , connections_to_ir_nodes_dict ) elif ir_node . node_type in [ 'TupleUnpack' , 'ListUnpack' ] : _handle_tuple_unpack_ir_node ( ir_node , connections_to_ir_nodes_dict ) elif ir_node . node_type in passthrough_ops : _handle_passthrough_ir_node ( ir_node , connections_to_ir_nodes_dict ) elif ir_node . node_type in input_ops_to_ignore : _handle_input_ir_node_to_ignore ( ir_node , connections_to_ir_nodes_dict )
Update input and output connections of certain ir_nodes in ir_nodes_list (Tuple/ListConstructs, passthrough,
input ops)
Update input and output connections of certain ir_nodes in ir_nodes_list ( Tuple / ListConstructs passthrough input ops )
"""
        Update input and output connections of certain ir_nodes in ir_nodes_list (Tuple/ListConstructs, passthrough,
        input ops)
        :param ir_nodes_list: List of ir_nodes to update connections for
        :param passthrough_ops: List of op types to treat as passthrough
        :param input_ops_to_ignore: List of input op types to ignore
        """
self
ir_nodes_list
List[IrNode]
List of ir_nodes to update connections for
List of ir_nodes to update connections for
passthrough_ops
List[str]
List of op types to treat as passthrough
List of op types to treat as passthrough
input_ops_to_ignore
List[str]
List of input op types to ignore
List of input op types to ignore