1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/coder/kernels/pmf_to_cdf_op_test.cc b/tensorflow/contrib/coder/kernels/pmf_to_cdf_op_test.cc
index c70e38faab..3408f6b519 100644
--- a/tensorflow/contrib/coder/kernels/pmf_to_cdf_op_test.cc
+++ b/tensorflow/contrib/coder/kernels/pmf_to_cdf_op_test.cc
@@ -82,7 +82,7 @@ class PmfToQuantizedCdfOpTest : public OpsTestBase {
         EXPECT_GT(diff, 0);
       }
 
-      EXPECT_LE(cdf_slice(cdf_slice.size() - 1), normalizer);
+      EXPECT_EQ(cdf_slice(cdf_slice.size() - 1), normalizer);
     }
   }
 };
@@ -98,6 +98,8 @@ TEST_F(PmfToQuantizedCdfOpTest, UnderSum) {
     GenerateData(&rand, {&matrix(i, 0), n});
   }
 
+  pmf.flat<float>() = pmf.flat<float>() * 0.85f;
+
   constexpr int kPrecision = 10;
   SetupOp(kPrecision, &pmf);
   TF_ASSERT_OK(RunOpKernel());
@@ -115,7 +117,7 @@ TEST_F(PmfToQuantizedCdfOpTest, OverSum) {
   matrix.setZero();
   const std::size_t n = matrix.dimension(1) / 2;
 
-  random::PhiloxRandom gen;
+  random::PhiloxRandom gen(random::New64(), random::New64());
   random::SimplePhilox rand(&gen);
   for (int64 i = 0; i < matrix.dimension(0); ++i) {
     GenerateData(&rand, {&matrix(i, 0), n});