diff options
author | 2018-03-12 10:34:34 -0700 | |
---|---|---|
committer | 2018-03-12 10:38:25 -0700 | |
commit | aab543c3013e3018d409ed2b8cd957f3465d1ab2 (patch) | |
tree | 036dd07e29f16c18cb19ee921994fd62317e4cb3 /tensorflow/contrib/lite/kernels/gemm_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/gemm_support.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/gemm_support.cc | 4 |
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; } |