华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
我看tbe算子源码,里面是没有形状推导的(至少是没有可以独立调用的形状推导功能),而AscendC编码的算子是有的(infershape)。
那么我的问题是,tbe算子是如何做形状推导的,它的形状推导功能是谁写的呢?
我要发帖子
我看tbe算子源码,里面是没有形状推导的(至少是没有可以独立调用的形状推导功能),而AscendC编码的算子是有的(infershape)。
那么我的问题是,tbe算子是如何做形状推导的,它的形状推导功能是谁写的呢?