diff options
author | 2018-07-06 14:50:08 -0700 | |
---|---|---|
committer | 2018-07-07 21:07:31 -0700 | |
commit | 2bfc7957dada57c1eb8491e04dac70d16b4369ac (patch) | |
tree | 2a967b9e5efbdb62f3e782fc9c4d8471fd78adc6 /tensorflow/contrib/lite/kernels/eigen_support.h | |
parent | 7c7cfde8fd6aa4ad7160b1fe6380e6007613d0d0 (diff) |
Make multithreaded conv respect setNumThreads()
PiperOrigin-RevId: 203527657
Diffstat (limited to 'tensorflow/contrib/lite/kernels/eigen_support.h')
-rw-r--r-- | tensorflow/contrib/lite/kernels/eigen_support.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/eigen_support.h b/tensorflow/contrib/lite/kernels/eigen_support.h index d47e691123..ec77856b10 100644 --- a/tensorflow/contrib/lite/kernels/eigen_support.h +++ b/tensorflow/contrib/lite/kernels/eigen_support.h @@ -17,6 +17,10 @@ limitations under the License. #include "tensorflow/contrib/lite/context.h" +namespace EigenForTFLite { +class ThreadPoolDevice; +} + namespace tflite { namespace eigen_support { @@ -28,6 +32,9 @@ void IncrementUsageCounter(TfLiteContext* context); // usages all temporary Eigen objects will be deleted. void DecrementUsageCounter(TfLiteContext* context); +const EigenForTFLite::ThreadPoolDevice* GetThreadPoolDevice( + TfLiteContext* context); + } // namespace eigen_support } // namespace tflite |