code: def _get_memory_profile(memory_profile_results):
      memory_profile = {}
      for line in memory_profile_results:
          if line.startswith("Memory:"):
              device_id = line.split()[1]
              avg_time_memory_alloc = float(line.split()[-1])
              memory_profile[
                  "max_storage_mem_alloc_" + device_id
              ] = avg_time_memory_alloc
  
      return memory_profile
  
file_path: mxnet/benchmark/opperf/utils/profiler_utils.py
imports: import time
  import functools
  import numpy
  from .common_utils import merge_map_list
  from mxnet import profiler
label_desc: Use image node_id 1 for calling a global function with example usage: _get_memory_profile(memory_profile_results) and returns: memory_profile