code: def _get_model_or_module(crypten_model):
      """
      Returns `Module` if model contains only one module. Otherwise returns model.
      """
      num_modules = len(list(crypten_model.modules()))
      if num_modules == 1:
          for crypten_module in crypten_model.modules():
              return crypten_module
      return crypten_model
  
imports: import copy
  import io
  import onnx
  import torch
  import torch.onnx.symbolic_helper
  import torch.onnx.utils
  from onnx import numpy_helper
  from torch.onnx import OperatorExportTypes
  from .None import module
label_desc: Use image node_id 9 for calling a global function with example usage: _get_model_or_module(crypten_model) and returns: crypten_model, crypten_module