daf8b438b0cc1819f234db6fb85465c6e0896f16,example/extensions/lib_pass/test_pass.py,,test_model,#Any#,50

Before Change

    aux = {}
    mysym2 = sym.optimize_for(pass_name,args,aux)
    print(mysym2.tojson())
    exe2 = mysym2.bind(ctx=mx.cpu(), args=args)
    out2 = exe2.forward()
    print(out2)

    // without propogating shapes/types
    print("-------------------------------")
    print("Testing pass "%s" without shapes/types" % pass_name)
    mysym3 = sym.optimize_for(pass_name, myOpt="yello")
    exe3 = mysym3.bind(ctx=mx.cpu(), args=args)
    out3 = exe3.forward()
    print(out3)

    // Gluon Hybridize
    print("-------------------------------")

After Change

    // execute in MXNet
    print("-------------------------------")
    print("Testing regular MXNet execution")
    inputs = [a,b]
    sym_block = nn.SymbolBlock(sym, inputs)
    sym_block.initialize()
    out = sym_block(mx.nd.ones((3,2)),mx.nd.ones((3,2)))
    print(out)

Instances