Python get_net示例

示例#1

def test_complexity():
    g = inception_v3.get_net(1, 1000, (3, 299, 299), "float32")
    run_opt_pass(g, transform.DeadCodeElimination())

示例#2

文件： test_pass_dead_code_elimination.py 项目： OctoML-NN-team/tvm

def test_complexity():
    mod = transform.InferType()(tvm.IRModule.from_expr(
        inception_v3.get_net(1, 1000, (3, 299, 299), "float32")))

    optimize_and_check(mod, mod,
                       transform.DeadCodeElimination(inline_once=True))