rename

x2paddle/op_mapper/pytorch2paddle/pytorch_op_mapper.py

@@ -16,7 +16,7 @@ import torch
 import numpy as np
 from x2paddle.core.op_mapper import OpMapper
 from x2paddle.core.util import *
-from x2paddle.core.paddle_graph import PaddleGraph
+from x2paddle.core.program import PaddleGraph
 from x2paddle.op_mapper.pytorch2paddle import prim
 from x2paddle.op_mapper.pytorch2paddle import aten
 

x2paddle/optimizer/linear_pass.py

@@ -14,7 +14,7 @@
 
 import numpy as np
 from x2paddle.core.util import *
-from x2paddle.core.paddle_graph import PaddleLayer, PaddleGraph
+from x2paddle.core.program import PaddleLayer, PaddleGraph
 from x2paddle.optimizer.passes import Pass, Matcher, PyTorchMatcher
 
 

x2paddle/optimizer/optimizer.py

@@ -45,5 +45,5 @@ class GraphOptimizer(object):
             self.current_pass = _pass
             self.current_matcher = matcher
             graph, _ = self.run(graph)
-            print("{} done!".format(pa.__class__.__name__))
+            print("{} done!".format(_pass.__class__.__name__))
         return graph

x2paddle/optimizer/passes.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from x2paddle.core.paddle_graph import PaddleGraph
+from x2paddle.core.program import PaddleGraph
 
 
 class Pass(object):