code: def get_memory_usage(assignments):
      ret = 0
      for cur in assignments:
          ret += _get_max_size(cur)
      return ret
  
imports: import networkx
  import collections
  import time
  import copy
  from caffe2.python import workspace, core
  from caffe2.proto import caffe2_pb2
  import enum
  import logging
  import caffe2.python._import_c_extension
label_desc: Use image node_id 19 for calling a global function with example usage: get_memory_usage(assignments) and returns: ret