aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/gemm_support.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-12 10:34:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-12 10:38:25 -0700
commitaab543c3013e3018d409ed2b8cd957f3465d1ab2 (patch)
tree036dd07e29f16c18cb19ee921994fd62317e4cb3 /tensorflow/contrib/lite/kernels/gemm_support.cc
parent31af33430d3edbfdecbcf121681e5a586f37ba03 (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/gemm_support.cc')
-rw-r--r--tensorflow/contrib/lite/kernels/gemm_support.cc4
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/gemm_support.cc b/tensorflow/contrib/lite/kernels/gemm_support.cc
index df8a9c8cee..76a5165d14 100644
--- a/tensorflow/contrib/lite/kernels/gemm_support.cc
+++ b/tensorflow/contrib/lite/kernels/gemm_support.cc
@@ -29,7 +29,9 @@ void IncrementUsageCounter(TfLiteContext* context) {
if (ptr == nullptr) {
ptr = new RefCountedGemmContext;
ptr->gemm_context_ = new gemmlowp::GemmContext();
- ptr->gemm_context_->set_max_num_threads(context->recommended_num_threads);
+ if (context->recommended_num_threads != -1) {
+ ptr->gemm_context_->set_max_num_threads(context->recommended_num_threads);
+ }
ptr->num_references_ = 0;
context->gemm_context = ptr;
}