将TensorFlow OpKernel固定到特定内核

Question

我写了OpKernel这是昂贵且有状态的。使用Eigen的NonBlockingThreadPool的默认实现和此线程池实现中的标准调度意味着

• OpKernel在任何可用的线程/核心上运行
• 此操作的状态必须转移到新核心，这会导致非最佳缓存行为

有没有办法将昂贵的操作固定在特定内核上运行？

Answer 1

目前还不行，但你并不是第一个有类似需求的人。