Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
39534800
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
大约 1 年 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
39534800
编写于
3月 14, 2022
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
deal with comments
上级
6ecd82a4
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
14 addition
and
25 deletion
+14
-25
x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py
x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py
+14
-25
未找到文件。
x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py
浏览文件 @
39534800
...
@@ -768,6 +768,8 @@ class CaffeOpMapper():
...
@@ -768,6 +768,8 @@ class CaffeOpMapper():
node
.
data
[
1
]).
astype
(
"float32"
)
node
.
data
[
1
]).
astype
(
"float32"
)
params
=
node
.
layer
.
scale_param
params
=
node
.
layer
.
scale_param
axis
=
params
.
axis
axis
=
params
.
axis
if
axis
<
0
:
axis
+=
len
(
node
.
in_shapes
[
0
])
if
len
(
node
.
inputs
)
==
2
:
if
len
(
node
.
inputs
)
==
2
:
input0
=
self
.
graph
.
get_input_node
(
node
,
idx
=
0
,
copy
=
True
)
input0
=
self
.
graph
.
get_input_node
(
node
,
idx
=
0
,
copy
=
True
)
input1
=
self
.
graph
.
get_input_node
(
node
,
idx
=
1
,
copy
=
True
)
input1
=
self
.
graph
.
get_input_node
(
node
,
idx
=
1
,
copy
=
True
)
...
@@ -776,11 +778,6 @@ class CaffeOpMapper():
...
@@ -776,11 +778,6 @@ class CaffeOpMapper():
inputs_dict
=
{}
inputs_dict
=
{}
inputs_dict
[
'x'
]
=
input0_name
inputs_dict
[
'x'
]
=
input0_name
inputs_dict
[
'y'
]
=
input1_name
inputs_dict
[
'y'
]
=
input1_name
self
.
paddle_graph
.
add_layer
(
"paddle.fluid.layers.elementwise_mul"
,
inputs
=
inputs_dict
,
outputs
=
[
node
.
layer_name
+
"_mul"
],
axis
=
1
)
else
:
else
:
self
.
paddle_graph
.
add_layer
(
self
.
paddle_graph
.
add_layer
(
"self.create_parameter"
,
"self.create_parameter"
,
...
@@ -793,13 +790,7 @@ class CaffeOpMapper():
...
@@ -793,13 +790,7 @@ class CaffeOpMapper():
inputs_dict
=
{}
inputs_dict
=
{}
inputs_dict
[
'x'
]
=
input0_name
inputs_dict
[
'x'
]
=
input0_name
inputs_dict
[
'y'
]
=
node
.
layer_name
+
"_cparam1"
inputs_dict
[
'y'
]
=
node
.
layer_name
+
"_cparam1"
if
len
(
node
.
in_shapes
[
0
])
==
2
:
if
axis
==
len
(
node
.
in_shapes
[
0
])
-
1
:
self
.
paddle_graph
.
add_layer
(
"paddle.multiply"
,
inputs
=
inputs_dict
,
outputs
=
[
node
.
layer_name
+
"_mul"
])
else
:
if
axis
==
-
1
or
axis
==
len
(
node
.
in_shapes
[
0
])
-
1
:
self
.
paddle_graph
.
add_layer
(
self
.
paddle_graph
.
add_layer
(
"paddle.multiply"
,
"paddle.multiply"
,
inputs
=
inputs_dict
,
inputs
=
inputs_dict
,
...
@@ -820,12 +811,10 @@ class CaffeOpMapper():
...
@@ -820,12 +811,10 @@ class CaffeOpMapper():
inputs_dict
[
'x'
]
=
node
.
layer_name
+
"_mul"
inputs_dict
[
'x'
]
=
node
.
layer_name
+
"_mul"
inputs_dict
[
'y'
]
=
node
.
layer_name
+
"_cparam2"
inputs_dict
[
'y'
]
=
node
.
layer_name
+
"_cparam2"
output_shape
=
node
.
out_shapes
[
0
]
output_shape
=
node
.
out_shapes
[
0
]
if
axis
==
-
1
:
if
axis
==
len
(
output_shape
)
-
1
:
self
.
paddle_graph
.
add_layer
(
self
.
paddle_graph
.
add_layer
(
"paddle.add"
,
inputs
=
inputs_dict
,
outputs
=
[
node
.
layer_name
])
"paddle.add"
,
inputs
=
inputs_dict
,
outputs
=
[
node
.
layer_name
])
else
:
else
:
if
axis
<
0
:
axis
=
axis
+
len
(
output_shape
)
param2_shape
=
self
.
params
[
node
.
layer_name
+
"_cparam2"
].
shape
param2_shape
=
self
.
params
[
node
.
layer_name
+
"_cparam2"
].
shape
param2_shape_len
=
len
(
param2_shape
)
param2_shape_len
=
len
(
param2_shape
)
diff_len
=
len
(
output_shape
)
-
axis
-
param2_shape_len
diff_len
=
len
(
output_shape
)
-
axis
-
param2_shape_len
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录