code: def fix_BoxWithNMSLimit(net):
      outputs = set()
      for op in net.op:
          if op.type == "BoxWithNMSLimit":
              outputs.add(op.output[0])
              outputs.add(op.output[1])
              outputs.add(op.output[2])
      for op in net.op:
          if op.type == "CopyIDEEPToCPU":
              if op.input[0] in outputs:
                  print(
                      "Chaning CopyIDEEPToCPU to Copy for {}".format(
                          op.input[0]
                      )
                  )
                  op.type = "Copy"
                  op.device_option.device_type = caffe2_pb2.CPU
  
file_path: pytorch/caffe2/python/mkl/rewrite_graph.py
imports: import copy
  from caffe2.proto import caffe2_pb2
  from caffe2.python import core
label_desc: Use image node_id 3 for calling a global function with example usage: fix_BoxWithNMSLimit(net) without return types