Python relay_mlpの例

コード例 #1

ファイル: test_graph.py プロジェクト: shjwudp/training_results_v0.7

def test_mlp():
    shape = {"data": (1, 1, 28, 28)}
    mx_fun = model_zoo.mx_mlp()
    mod, _ = relay.frontend.from_mxnet(mx_fun, shape=shape)
    relay_fun = model_zoo.relay_mlp()
    compare_graph(mod, relay_fun)

コード例 #2

ファイル: test_graph.py プロジェクト: LANHUIYING/tvm

def test_mlp():
    shape = {"data": (1, 1, 28, 28)}
    mx_fun = model_zoo.mx_mlp()
    from_mx_fun, _ = relay.frontend.from_mxnet(mx_fun, shape=shape)
    relay_fun = model_zoo.relay_mlp()
    compare_graph(from_mx_fun, relay_fun)