def METHOD_NAME(self, prunable_layers_types) -> Dict:
    retval = set()
    for node in self._graph.get_all_nodes():
        if node.node_type in prunable_layers_types and not (
            is_grouped_conv(node) or is_batched_linear(node, self._graph)
        ):
            retval.add(node.node_id)
    return retval