code: def build_graph():
      data = mx.sym.var("data")
      data = mx.sym.flatten(data=data)
      fc1 = mx.sym.FullyConnected(data=data, num_hidden=128)
      act1 = mx.sym.Activation(data=fc1, act_type="relu")
      fc2 = mx.sym.FullyConnected(data=act1, num_hidden=64)
      act2 = mx.sym.Activation(data=fc2, act_type="relu")
      fc3 = mx.sym.FullyConnected(data=act2, num_hidden=10)
      return mx.sym.SoftmaxOutput(data=fc3, name="softmax")
  
file_path: sagemaker-python-sdk/tests/data/mxnet_mnist/mnist_neo.py
imports: from __future__ import absolute_import
  import argparse
  import gzip
  import json
  import logging
  import os
  import struct
  import mxnet
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: build_graph() and returns: mx