code: def parse_init_json(data):
      params = {}
      for key in data:
          value = data[key]
          layer_name = value["_name"]
          if layer_name == "Empty":
              # Empty Layer
              params[key] = ["Empty"]
          elif layer_name == "Conv":
              # Conv layer
              params[key] = [
                  layer_name,
                  value["kernel_size"],
                  value["kernel_size"],
              ]
          else:
              # Pooling Layer
              params[key] = [
                  layer_name,
                  value["pooling_size"],
                  value["pooling_size"],
              ]
      return params
  
imports: from __future__ import absolute_import, division, print_function
  import logging
  import math
  import tempfile
  import time
  import argparse
  import tensorflow
  from tensorflow.examples.tutorials.mnist import input_data
  import nni
label_desc: Use image node_id 4 for calling a global function with example usage: parse_init_json(data) and returns: params