diff options
author | 2018-03-26 11:47:50 -0700 | |
---|---|---|
committer | 2018-03-26 11:50:29 -0700 | |
commit | d2604f8dcb8a63ca063f712c24ce5aa63403b0aa (patch) | |
tree | 6109456e2238ab20f647639aa8f05c0aba5b128d /tensorflow/contrib/lite/kernels/gemm_support.h | |
parent | 6d46c21e9f300d07e30a2185671f07d34fac3999 (diff) |
Revert to initializing number of threads when SetNumThreads is called. Requiring it
to happen before OpInit() is way too confusing for users.
PiperOrigin-RevId: 190499644
Diffstat (limited to 'tensorflow/contrib/lite/kernels/gemm_support.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/gemm_support.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/gemm_support.h b/tensorflow/contrib/lite/kernels/gemm_support.h index 37af772c68..f033501cb6 100644 --- a/tensorflow/contrib/lite/kernels/gemm_support.h +++ b/tensorflow/contrib/lite/kernels/gemm_support.h @@ -45,6 +45,9 @@ void IncrementUsageCounter(TfLiteContext* context); // 'context'. If there are no more usages the GemmContext will be deleted. void DecrementUsageCounter(TfLiteContext* context); +// Set the number of threads that can be used by gemmlowp. +void SetNumThreads(TfLiteContext* context, int num_threads); + } // namespace gemm_support } // namespace tflite |