From 9235dcb638617c666cc4d14ea770131c3b767173 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 14 Jun 2017 16:43:57 -0700
Subject: [PATCH] Fix a flaky test in graph_transforms:transforms_test

PiperOrigin-RevId: 159044391
---
 tensorflow/tools/graph_transforms/quantize_nodes_test.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tensorflow/tools/graph_transforms/quantize_nodes_test.cc b/tensorflow/tools/graph_transforms/quantize_nodes_test.cc
index f070039381c..eca263a1ae0 100644
--- a/tensorflow/tools/graph_transforms/quantize_nodes_test.cc
+++ b/tensorflow/tools/graph_transforms/quantize_nodes_test.cc
@@ -672,8 +672,7 @@ class QuantizeNodesTest : public ::testing::Test {
     TF_ASSERT_OK(root.ToGraphDef(&float_graph_def));
 
     Tensor input_tensor(DT_FLOAT, {1, 128, 128, 3});
-    auto float_input_tensor = input_tensor.flat<float>();
-    float_input_tensor.constant(100.0f);
+    test::FillFn<float>(&input_tensor, [](int) { return 100.0f; });
 
     TestQuantizedVersusFloatGraph(float_graph_def,
                                   {{"placeholder_op", input_tensor}},