fix test_memory_optimization_transpiler.py

7 years ago · 483f5e7efd
parent 345737d0fe
commit 483f5e7efd
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/test_memory_optimization_transpiler.py
+++ b/python/paddle/fluid/tests/unittests/test_memory_optimization_transpiler.py
@ -18,7 +18,7 @@ import unittest
 import paddle.fluid.layers as layers
 import paddle.fluid.optimizer as optimizer
 from paddle.fluid.framework import Program, program_guard
-from paddle.fluid.memory_optimization_transpiler import memory_optimize
+from paddle.fluid.transpiler import memory_optimize


 class TestControlFlowGraph(unittest.TestCase):