| 87 | } |
| 88 | |
| 89 | int GetWorkersCount(OpKernelContext* tf_context) { |
| 90 | if (g_num_threads == 0) { |
| 91 | return tf_context->device()->tensorflow_cpu_worker_threads()->num_threads; |
| 92 | } |
| 93 | return g_num_threads; |
| 94 | } |
| 95 | |
| 96 | typedef gemmlowp::meta::SimpleContext<gemmlowp::WorkersPool> LocalContext; |
| 97 |
no test coverage detected