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

Update tensorflow/core/kernels/mkl_quantized_conv_ops.h

Browse Source 
Co-authored-by: Penporn Koanantakool <38085909+penpornk@users.noreply.github.com>
This commit is contained in:
Sharada Shiddibhavi 2020-06-08 10:30:03 -07:00 committed by GitHub
parent e2b4994f45
commit 7fc6bf6726
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/mkl_quantized_conv_ops.h
View File

@ -72,7 +72,7 @@ void MklQuantizationRangeForMultiplication(float min_a, float max_a,
#ifndef ENABLE_MKLDNN_THREADPOOL
#pragma omp parallel for
#endif // ENABLE_MKLDNN_THREADPOOL
#endif // !ENABLE_MKLDNN_THREADPOOL
// TODO: Add eigen parallel_for
for (size_t n = 0; n < n_channel; ++n) {
float a_float_for_one_quant_level =