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

Fix a flaky test in graph_transforms:transforms_test

Browse Source 
PiperOrigin-RevId: 159044391
This commit is contained in:
A. Unique TensorFlower 2017-06-14 16:43:57 -07:00 committed by TensorFlower Gardener
parent f386c88501
commit 9235dcb638
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/tools/graph_transforms/quantize_nodes_test.cc
View File

@ -672,8 +672,7 @@ class QuantizeNodesTest : public ::testing::Test {
TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); TF_ASSERT_OK(root.ToGraphDef(&float_graph_def));
Tensor input_tensor(DT_FLOAT, {1, 128, 128, 3}); Tensor input_tensor(DT_FLOAT, {1, 128, 128, 3});
auto float_input_tensor = input_tensor.flat<float>(); test::FillFn<float>(&input_tensor, [](int) { return 100.0f; });
float_input_tensor.constant(100.0f);
TestQuantizedVersusFloatGraph(float_graph_def, TestQuantizedVersusFloatGraph(float_graph_def,
{{"placeholder_op", input_tensor}}, {{"placeholder_op", input_tensor}},