nncf/experimental/torch/search_building_blocks/search_blocks.py
def add_node_to_aux_struct(node: SearchGraphNode, shape: List[int], shape_map: ShapeVsNodesMap):
    """
    Add to shape_map key of node for corresponds shape.
    """
    str_shape = str(shape)
    if str_shape in shape_map:
        shape_map[str_shape].add(node)
    else:
        shape_map[str_shape] = {node}

    Add to shape_map key of node for corresponds shape.
    
Add to shape_map key of node for corresponds shape.
Add to shape_map key of node for corresponds shape .
def add_node_to_aux_struct(node: SearchGraphNode, shape: List[int], shape_map: ShapeVsNodesMap):
    str_shape = str(shape)
    if str_shape in shape_map:
        shape_map[str_shape].add(node)
    else:
        shape_map[str_shape] = {node}
def add_node_to_aux_struct ( node : SearchGraphNode , shape : List [ int ] , shape_map : ShapeVsNodesMap ) : str_shape = str ( shape ) if str_shape in shape_map : shape_map [ str_shape ] . add ( node ) else : shape_map [ str_shape ] = { node }
Add to shape_map key of node for corresponds shape.
Add to shape_map key of node for corresponds shape .
"""
    Add to shape_map key of node for corresponds shape.
    """
node
SearchGraphNode
shape
List[int]
shape_map
ShapeVsNodesMap