首页 > 解决方案 > 如何获取 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 线程数?

标签: tensorfloweigeneigen3

解决方案


推荐阅读