TensorFlow Lite中的线程/线程池

时间:2019-07-19 13:28:54

标签: tensorflow tensorflow-lite

我正在查看tensorflow中的AvgPoolGrad op,以了解如何在tensorflow lite中移植代码。

我在TF中注意到代码使用线程池,所以我在想tflite中有类似的东西吗?

不仅要在特定的自定义操作中使用它,因为tflite不支持AvgPoolGrad,而且总的来说,如果我想创建自己的自定义操作并希望将工作负载分配到不同的线程上以加速数学运算,怎么办? / p>

有样品吗?

thnx

0 个答案:

没有答案