Toybrick

Tensorflow 算子支持请求

程子

中级会员

积分
386
发表于 2019-2-18 10:34:01    查看: 18970|回复: 15 | [复制链接]    打印 | 显示全部楼层
本帖最后由 程子 于 2019-3-4 15:42 编辑

ExpandDims
Maximum
Minimum
RealDiv
SparseToDense
Sqrt
Sub
TensorArrayGatherV3
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2019-2-18 12:49:19 | 显示全部楼层
问题已经上报给相关部门了,有进展我们会发布新版本的rknn-toolkit
回复

使用道具 举报

程子

中级会员

积分
386
 楼主| 发表于 2019-2-18 13:48:44 | 显示全部楼层
大约多久可以完成?
回复

使用道具 举报

程子

中级会员

积分
386
 楼主| 发表于 2019-2-19 17:45:02 | 显示全部楼层
顺便问一下,有没有目前支持的算子列表?
回复

使用道具 举报

程子

中级会员

积分
386
 楼主| 发表于 2019-2-21 14:37:32 | 显示全部楼层
更新了不支持的算子列表:
ExpandDims
Maximum
Minimum
RealDiv
Sqrt
Sub
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2019-2-22 10:43:10 | 显示全部楼层
非常感谢,我们也在自测不支持的op然后催相关人员处理。我目前临时做法是不支持的op放在3399那端计算,然后接入Rknn,把模型拆解用。当然太复杂的模型就不适用了
回复

使用道具 举报

程子

中级会员

积分
386
 楼主| 发表于 2019-2-22 10:55:34 | 显示全部楼层
【不支持的op放在3399那端计算,然后接入Rknn,把模型拆解用】
这个操作我们自己可以实现吗?如果可以,有没有教程啥的?
回复

使用道具 举报

程子

中级会员

积分
386
 楼主| 发表于 2019-2-22 12:08:43 | 显示全部楼层
能否给出一个切分网络的例子?比如以mobilenet-ssd为例,中间抽出来一个OP给CPU做,其他用NPU做。看看最终结果对性能有多大影响?
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2019-2-23 18:03:32 | 显示全部楼层
程子 发表于 2019-2-22 12:08
能否给出一个切分网络的例子?比如以mobilenet-ssd为例,中间抽出来一个OP给CPU做,其他用NPU做。看看最终 ...

看你权重文件怎么写的了,有些语言权重是可以单个节点load进去的,网络就非常好拆。
要是都是下现成pb文件,那拆起来是有点难度了。。
我也只是自己在试着这么玩,绕过不支持的算子。
回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2019-2-23 18:04:22 | 显示全部楼层
我觉得大家可以一起统计下不支持的OP列表,我会报给相关人员一并处理
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表