code: def set_flags(_enabled):
      orig_flags = (torch._C._get_mkldnn_enabled(),)
      torch._C._set_mkldnn_enabled(_enabled)
      return orig_flags
  
file_path: pytorch/torch/backends/mkldnn/__init__.py
imports: import sys
  from contextlib import contextmanager
  import torch
  from torch.backends import __allow_nonbracketed_mutation, ContextProp, PropModule
label_desc: Use image node_id 2 for calling a global function with example usage: set_flags(_enabled) and returns: orig_flags