如何获得 inter_op 线程编号在 Tensorflow Op 内核中

问题描述

我试图在 Tensorflow Op 内核 esp 中获取 inter_op 线程号。核的计算函数
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/concat_op.cc#L62

比如如果我们给 INTER_OP 的值大于 1,那么在 Op Kernel 中我们怎么知道 inter_op 线程号?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)