diff options
Diffstat (limited to 'tensorflow/python/tools/optimize_for_inference_test.py')
| -rw-r--r-- | tensorflow/python/tools/optimize_for_inference_test.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/python/tools/optimize_for_inference_test.py b/tensorflow/python/tools/optimize_for_inference_test.py index e547d0e292..447057cfe9 100644 --- a/tensorflow/python/tools/optimize_for_inference_test.py +++ b/tensorflow/python/tools/optimize_for_inference_test.py @@ -271,7 +271,8 @@ class OptimizeForInferenceTest(test.TestCase): for node in optimized_graph_def.node: self.assertNotEqual("Conv2D", node.op) - self.assertNotEqual("ResizeBilinear", node.op) + self.assertNotEqual("MirrorPad", node.op) + def testFusePadAndConv(self): with self.test_session() as sess: @@ -299,7 +300,7 @@ class OptimizeForInferenceTest(test.TestCase): for node in optimized_graph_def.node: self.assertNotEqual("Conv2D", node.op) - self.assertNotEqual("MirrorPad", node.op) + self.assertNotEqual("ResizeBilinear", node.op) if __name__ == "__main__": |