experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Avoid using Eigen ThreadPool when thread count is 1

Browse Source 
PiperOrigin-RevId: 239232993
This commit is contained in:
Jared Duke 2019-03-19 11:28:41 -07:00 committed by TensorFlower Gardener
parent 067a3a6c3f
commit eb1d8f7954
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/lite/kernels/conv.cc
View File

@ -526,15 +526,19 @@ void EvalFloat(TfLiteContext* context, TfLiteNode* node,
float output_activation_min, output_activation_max;
CalculateActivationRange(params->activation, &output_activation_min,
&output_activation_max);
KernelType effective_kernel_type;
if ((kernel_type == kMultithreadOptimized) &&
(params->dilation_width_factor != 1 ||
params->dilation_height_factor != 1)) {
KernelType effective_kernel_type = kernel_type;
if (kernel_type == kMultithreadOptimized) {
if (context->recommended_num_threads == 1) {
// Use of kMultithreadOptimized is precomputed during |Prepare()|, whereas
// the actual thread count can change at any time. If the client requests
// a single thread (after Prepare()), fall back to optimized.
effective_kernel_type = kGenericOptimized;
} else if ((params->dilation_width_factor != 1) ||
(params->dilation_height_factor != 1)) {
// kMultithreadOptimized does not support dilation.
// Therefore, fallback to optimized.
effective_kernel_type = kGenericOptimized;
} else {
effective_kernel_type = kernel_type;
}
}
ConvParams op_params;
op_params.padding_type = RuntimePaddingType(params->padding);