From 300d9c7181a7a256c733564ec3aeb33a8b76b2bc Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Fri, 24 May 2019 10:14:49 -0700
Subject: [PATCH] Fix flaky eigen_mkldnn_contraction_kernel test

PiperOrigin-RevId: 249853404
---
 tensorflow/core/kernels/eigen_mkldnn_contraction_kernel_test.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/core/kernels/eigen_mkldnn_contraction_kernel_test.cc b/tensorflow/core/kernels/eigen_mkldnn_contraction_kernel_test.cc
index 3a1ea7da43c..cbf525212bb 100644
--- a/tensorflow/core/kernels/eigen_mkldnn_contraction_kernel_test.cc
+++ b/tensorflow/core/kernels/eigen_mkldnn_contraction_kernel_test.cc
@@ -166,7 +166,7 @@ TEST(EigenMkldnnTest, MkldnnGemmQInt8xQUInt8) {
   rhs.setRandom();
   // NOTE: 's8*u8 + s8*u8 -> s16' saturation might lead to incorrect results. In
   // 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;
   contract_dims[0].first = 1;