def METHOD_NAME(graph: NNCFGraph, nodes: List):
    """
    Initialize masks in graph for mask propagation algorithm
    :param graph: NNCFNetwork
    :param nodes: list with pruned nodes
    """
    for node in graph.get_all_nodes():
        node.attributes.pop("output_mask", None)
    for minfo in nodes:
        mask = minfo.operand.binary_filter_pruning_mask
        nncf_node = graph.get_node_by_id(minfo.nncf_node_id)
        nncf_node.attributes["output_mask"] = PTNNCFTensor(mask)