def valid_consumers(self, name: str, dag: OnnxDAG) -> bool:
        """Check if the consumers of the node are valid.

        This is to prevent checking the tensor value of large tensors unnecessarily.
        """
        for consumer in dag.get_consumers(name):
            if dag.get_node_op_type(consumer) not in ReplaceAttentionMaskValue.ALLOWED_CONSUMER_OPS:
                return False
        return True
Check if the consumers of the node are valid.

        This is to prevent checking the tensor value of large tensors unnecessarily.
        
olive/passes/onnx/graph_surgeries.py