Yaoyao Ding yaoyaoding
yaoyaoding
/ main.py
Created
January 1, 2020 05:11
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import numpy as np | |
| import argparse | |
| import os | |
| from typing import Tuple | |
| import io | |
| import tvm | |
| import tvm.relay | |
| import time | |
| import cv2 | |
| import torch |
yaoyaoding
/ demo.py
Last active
October 9, 2019 20:45
The code to reproduce error using the auto-tuned schedule, which is modified from the auto-tune tutorial.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import numpy as np | |
| import tvm | |
| from tvm import autotvm | |
| from tvm import relay | |
| import tvm.relay.testing | |
| from tvm.autotvm.tuner import XGBTuner, GATuner, RandomTuner, GridSearchTuner | |
| from tvm.contrib.util import tempdir | |
| import tvm.contrib.graph_runtime as runtime |
NewerOlder