Toybrick

标题: Tensorflow 算子支持请求 [打印本页]

作者: 程子 时间: 2019-2-18 10:34
标题: Tensorflow 算子支持请求
本帖最后由程子于 2019-3-4 15:42 编辑

ExpandDims
Maximum
Minimum
RealDiv
SparseToDense
Sqrt
Sub
TensorArrayGatherV3

作者: jefferyzhang 时间: 2019-2-18 12:49
问题已经上报给相关部门了，有进展我们会发布新版本的rknn-toolkit

作者: 程子 时间: 2019-2-18 13:48
大约多久可以完成？

作者: 程子 时间: 2019-2-19 17:45
顺便问一下，有没有目前支持的算子列表？

作者: 程子 时间: 2019-2-21 14:37
更新了不支持的算子列表：
ExpandDims
Maximum
Minimum
RealDiv
Sqrt
Sub

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

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

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

作者: jefferyzhang 时间: 2019-2-23 18:03

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

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

作者: jefferyzhang 时间: 2019-2-23 18:04
我觉得大家可以一起统计下不支持的OP列表，我会报给相关人员一并处理

作者: shanshuiziran 时间: 2019-2-25 17:43

程子发表于 2019-2-21 14:37
更新了不支持的算子列表：
ExpandDims
Maximum

请问，更新完了再哪里下载？

作者: 程子 时间: 2019-2-25 18:18

shanshuiziran 发表于 2019-2-25 17:43
请问，更新完了再哪里下载？

我觉得你要不重新看一遍帖子。。。

作者: 程子 时间: 2019-2-27 10:43
更新一个不支持的算子：
SparseToDense

作者: 程子 时间: 2019-3-4 15:43
更新一个不支持的算子：
TensorArrayGatherV3

作者: kimizij 时间: 2019-3-7 14:50

jefferyzhang 发表于 2019-2-23 18:04
我觉得大家可以一起统计下不支持的OP列表，我会报给相关人员一并处理

是不是tf.add也不支持啊，我自己的模型里有tf.add就转换失败了

作者: jefferyzhang 时间: 2019-5-21 14:37

kimizij 发表于 2019-3-7 14:50
是不是tf.add也不支持啊，我自己的模型里有tf.add就转换失败了

硬件op和软件写法不一样的，需要配合其他过程组成一个完整的op。

欢迎光临 Toybrick (https://t.rock-chips.com/)