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

Fix flaky eigen_mkldnn_contraction_kernel test

Browse Source 
PiperOrigin-RevId: 249853404
This commit is contained in:
Eugene Zhulenev 2019-05-24 10:14:49 -07:00 committed by TensorFlower Gardener
parent 36147a36e5
commit 300d9c7181
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/eigen_mkldnn_contraction_kernel_test.cc
View File

@ -166,7 +166,7 @@ TEST(EigenMkldnnTest, MkldnnGemmQInt8xQUInt8) {
rhs.setRandom(); rhs.setRandom();
// NOTE: 's8*u8 + s8*u8 -> s16' saturation might lead to incorrect results. In // NOTE: 's8*u8 + s8*u8 -> s16' saturation might lead to incorrect results. In
// practice in FusedConv2DBiasActivationKernel we use 7 bit inputs. // practice in FusedConv2DBiasActivationKernel we use 7 bit inputs.
rhs.clip(0, 127); rhs = rhs.clip(0, 127);
Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims;
contract_dims[0].first = 1; contract_dims[0].first = 1;