diff options
author | 2018-03-23 16:16:22 -0700 | |
---|---|---|
committer | 2018-03-25 04:25:49 -0700 | |
commit | 97249979d9a76ae05d590f9cbe199c0b47712b4f (patch) | |
tree | dc3571d9421aa8f9ba75e4f8e9c9377885358536 | |
parent | f54f57337078c93877df5c9a1b126e879f5b33a5 (diff) |
bug fix: evaluate nodes before swap the original graph
PiperOrigin-RevId: 190291844
-rw-r--r-- | tensorflow/core/grappler/optimizers/constant_folding_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/grappler/optimizers/constant_folding_test.cc b/tensorflow/core/grappler/optimizers/constant_folding_test.cc index 914a9257ee..6340565bcd 100644 --- a/tensorflow/core/grappler/optimizers/constant_folding_test.cc +++ b/tensorflow/core/grappler/optimizers/constant_folding_test.cc @@ -1922,6 +1922,8 @@ TEST_F(ConstantFoldingTest, PartialFolding_Concat) { item.fetch = {"concat0", "concat1", "concat2", "concat3", "concat4", "concat5", "concat6", "concat7", "concat8", "concat9"}; + auto tensors_expected = EvaluateNodes(item.graph, {"concat0"}); + EXPECT_EQ(1, tensors_expected.size()); ConstantFolding optimizer(nullptr /* cpu_device */); GraphDef output; Status status = optimizer.Optimize(nullptr, item, &output); @@ -1971,9 +1973,7 @@ TEST_F(ConstantFoldingTest, PartialFolding_Concat) { } } - auto tensors_expected = EvaluateNodes(item.graph, {"concat0"}); auto tensors = EvaluateNodes(output, {"concat0"}); - EXPECT_EQ(1, tensors_expected.size()); EXPECT_EQ(1, tensors.size()); test::ExpectTensorNear<float>(tensors_expected[0], tensors[0], 1e-6); } |