code: def mobilenet_unit(
      data,
      num_filter=1,
      kernel=(1, 1),
      stride=(1, 1),
      pad=(0, 0),
      num_group=1,
      if_act=True,
      prefix="",
  ):
      conv = mx.sym.Convolution(
          data=data,
          num_filter=num_filter,
          kernel=kernel,
          num_group=num_group,
          stride=stride,
          pad=pad,
          no_bias=True,
          name="%s-conv2d" % prefix,
      )
      bn = mx.sym.BatchNorm(
          data=conv,
          name="%s-batchnorm" % prefix,
          fix_gamma=False,
          use_global_stats=False,
          eps=1e-5,
      )
      if if_act:
          act = relu6(bn, prefix)
          return act
      else:
          return bn
  
file_path: byteps/example/mxnet/symbols/mobilenetv2.py
imports: import mxnet
label_desc: Use image node_id 3 for calling a global function with example usage: mobilenet_unit(data, num_filter, kernel, stride, pad, num_group, if_act, prefix) and returns: act, bn