2024年4月7日
TVM Pass优化 -- InferType 类型推导
摘要： 定义（What） InferType，类型推断，顾名思义，给表达式进行类型的推断 直接上代码 import tvm from tvm import relay import numpy as np def get_demo_mod(): a = relay.var("a", shape=(2, 3, 阅读全文
posted @ 2024-04-07 20:31 牛犁heart 阅读(114) 评论(0) 推荐(0) 编辑