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

Update mkl_requantization_range_per_channel_op.cc

Browse Source
This commit is contained in:
Sharada Shiddibhavi 2020-06-08 10:31:00 -07:00 committed by GitHub
parent 7fc6bf6726
commit 7b4b07c098
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_requantization_range_per_channel_op.cc
View File

@ -77,7 +77,7 @@ class MklRequantizationRangePerChannelOp : public OpKernel {
#ifndef ENABLE_MKLDNN_THREADPOOL
#pragma omp parallel for reduction(max : out_min_max)
#endif // ENABLE_MKLDNN_THREADPOOL
#endif // !ENABLE_MKLDNN_THREADPOOL
// TODO: Add eigen parallel_for
for (size_t i = 0; i < depth; ++i) {
Eigen::Tensor<qint32, 0, Eigen::RowMajor> min =