Esempi in Python per get_net

Esempio n. 1

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

Esempio n. 2

File: test_pass_dead_code_elimination.py Progetto: 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))