From d8bb89202d90ce1ff8f8dea2cb2f1860669030c0 Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Tue, 11 Aug 2020 17:21:44 +0800 Subject: [PATCH] rename --- x2paddle/op_mapper/pytorch2paddle/pytorch_op_mapper.py | 2 +- x2paddle/optimizer/linear_pass.py | 2 +- x2paddle/optimizer/optimizer.py | 2 +- x2paddle/optimizer/passes.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/x2paddle/op_mapper/pytorch2paddle/pytorch_op_mapper.py b/x2paddle/op_mapper/pytorch2paddle/pytorch_op_mapper.py index 6284e19..4889fae 100644 --- a/x2paddle/op_mapper/pytorch2paddle/pytorch_op_mapper.py +++ b/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 diff --git a/x2paddle/optimizer/linear_pass.py b/x2paddle/optimizer/linear_pass.py index 6af4819..e0267f9 100644 --- a/x2paddle/optimizer/linear_pass.py +++ b/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 diff --git a/x2paddle/optimizer/optimizer.py b/x2paddle/optimizer/optimizer.py index c03b9e9..5b06e54 100644 --- a/x2paddle/optimizer/optimizer.py +++ b/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 diff --git a/x2paddle/optimizer/passes.py b/x2paddle/optimizer/passes.py index 028a211..8b80d8a 100644 --- a/x2paddle/optimizer/passes.py +++ b/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): -- GitLab