Tensorflow 算子支持请求

[复制链接] · 发表于 2019-2-18 10:34:01

本帖最后由程子于 2019-3-4 15:42 编辑

ExpandDims
Maximum
Minimum
RealDiv
SparseToDense
Sqrt
Sub
TensorArrayGatherV3

只看该作者 · 发表于 2019-2-18 12:49:19

问题已经上报给相关部门了，有进展我们会发布新版本的rknn-toolkit

只看该作者 · 发表于 2019-2-18 13:48:44

大约多久可以完成？

只看该作者 · 发表于 2019-2-19 17:45:02

顺便问一下，有没有目前支持的算子列表？

只看该作者 · 发表于 2019-2-21 14:37:32

更新了不支持的算子列表：
ExpandDims
Maximum
Minimum
RealDiv
Sqrt
Sub

只看该作者 · 发表于 2019-2-22 10:43:10

非常感谢，我们也在自测不支持的op然后催相关人员处理。我目前临时做法是不支持的op放在3399那端计算，然后接入Rknn，把模型拆解用。当然太复杂的模型就不适用了

只看该作者 · 发表于 2019-2-22 10:55:34

【不支持的op放在3399那端计算，然后接入Rknn，把模型拆解用】
这个操作我们自己可以实现吗？如果可以，有没有教程啥的？

只看该作者 · 发表于 2019-2-22 12:08:43

能否给出一个切分网络的例子？比如以mobilenet-ssd为例，中间抽出来一个OP给CPU做，其他用NPU做。看看最终结果对性能有多大影响？

只看该作者 · 发表于 2019-2-23 18:03:32

程子发表于 2019-2-22 12:08
能否给出一个切分网络的例子？比如以mobilenet-ssd为例，中间抽出来一个OP给CPU做，其他用NPU做。看看最终 ...

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

只看该作者 · 发表于 2019-2-23 18:04:22

我觉得大家可以一起统计下不支持的OP列表，我会报给相关人员一并处理