Skip to content
Snippets Groups Projects
Normal view Permalink
test_frontend.py 139 B
Newer Older
  • Learn to ignore specific revisions
    • Yifan Zhao's avatar
    Added dnn pytorch sources and simplest frontend test
    Yifan Zhao committed
    1 2 3 4 
    from torch2hpvm import compile_torch_module
from dnn import AlexNet

compile_torch_module(AlexNet(), (1, 3, 32, 32), "/tmp/alexnet", True)