diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-12 10:34:34 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-12 10:38:25 -0700 |
commit | aab543c3013e3018d409ed2b8cd957f3465d1ab2 (patch) | |
tree | 036dd07e29f16c18cb19ee921994fd62317e4cb3 /tensorflow/contrib/lite/kernels/eigen_support.cc | |
parent | 31af33430d3edbfdecbcf121681e5a586f37ba03 (diff) |
Make default number of threads trigger the default behavior for both eigen
and gemmlowp. In gemmlowp the default is '1', while in eigen it is 'number of
processors'.
PiperOrigin-RevId: 188742087
Diffstat (limited to 'tensorflow/contrib/lite/kernels/eigen_support.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/eigen_support.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/eigen_support.cc b/tensorflow/contrib/lite/kernels/eigen_support.cc index 1435a45672..213e465552 100644 --- a/tensorflow/contrib/lite/kernels/eigen_support.cc +++ b/tensorflow/contrib/lite/kernels/eigen_support.cc @@ -27,8 +27,9 @@ struct RefCountedEigenContext { void IncrementUsageCounter(TfLiteContext* context) { auto* ptr = reinterpret_cast<RefCountedEigenContext*>(context->eigen_context); if (ptr == nullptr) { - Eigen::setNbThreads(context->recommended_num_threads); - + if (context->recommended_num_threads != -1) { + Eigen::setNbThreads(context->recommended_num_threads); + } ptr = new RefCountedEigenContext; ptr->num_references = 0; context->eigen_context = ptr; |