Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
未来在这儿
X2Paddle
提交
cf75526c
X
X2Paddle
项目概览
未来在这儿
/
X2Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / X2Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
cf75526c
编写于
1月 20, 2022
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fixed SpaceToBatchND bug
上级
1690cfd1
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
12 addition
and
11 deletion
+12
-11
x2paddle/op_mapper/tf2paddle/tf_op_mapper.py
x2paddle/op_mapper/tf2paddle/tf_op_mapper.py
+12
-11
未找到文件。
x2paddle/op_mapper/tf2paddle/tf_op_mapper.py
浏览文件 @
cf75526c
...
...
@@ -18,7 +18,7 @@ from x2paddle.core.util import *
import
traceback
import
math
import
inspect
import
numpy
import
numpy
as
np
import
sys
name_counter
=
dict
()
...
...
@@ -416,8 +416,8 @@ class TFOpMapper():
kernel_value
=
self
.
decoder
.
infer_tensor
(
kernel
,
use_diff_inputs
=
False
)
kernel_weight_name
=
op_name
+
".weight"
self
.
params
[
kernel_weight_name
]
=
n
umpy
.
transpose
(
kernel_value
,
(
3
,
2
,
0
,
1
))
self
.
params
[
kernel_weight_name
]
=
n
p
.
transpose
(
kernel_value
,
(
3
,
2
,
0
,
1
))
input_name
=
input
.
name
if
data_format
==
"NHWC"
:
...
...
@@ -482,8 +482,8 @@ class TFOpMapper():
kernel_value
=
self
.
decoder
.
infer_tensor
(
kernel
,
use_diff_inputs
=
False
)
kernel_weight_name
=
op_name
+
".weight"
self
.
params
[
kernel_weight_name
]
=
n
umpy
.
transpose
(
kernel_value
,
(
4
,
3
,
0
,
1
,
2
))
self
.
params
[
kernel_weight_name
]
=
n
p
.
transpose
(
kernel_value
,
(
4
,
3
,
0
,
1
,
2
))
input_name
=
input
.
name
if
data_format
==
"NDHWC"
:
...
...
@@ -630,7 +630,7 @@ class TFOpMapper():
"shape"
:
param
.
name
},
outputs
=
[
node
.
name
])
if
param
.
layer_type
!=
"Const"
:
out_shape
=
n
umpy
.
array
(
node
.
out_shapes
[
0
])
out_shape
=
n
p
.
array
(
node
.
out_shapes
[
0
])
if
(
out_shape
>
0
).
any
():
out_shape
[
out_shape
<
0
]
=
0
self
.
paddle_graph
.
add_layer
(
...
...
@@ -774,8 +774,8 @@ class TFOpMapper():
pad_mode
=
node
.
get_attr
(
"padding"
).
decode
()
kernel_weight_name
=
op_name
+
".weight"
self
.
params
[
kernel_weight_name
]
=
n
umpy
.
transpose
(
kernel
.
value
,
(
2
,
3
,
0
,
1
))
self
.
params
[
kernel_weight_name
]
=
n
p
.
transpose
(
kernel
.
value
,
(
2
,
3
,
0
,
1
))
input_name
=
input
.
name
if
data_format
==
"NHWC"
:
...
...
@@ -1304,7 +1304,7 @@ class TFOpMapper():
data_format
=
node
.
get_attr
(
"data_format"
).
decode
()
kernel_name
=
op_name
+
".weight"
self
.
params
[
kernel_name
]
=
n
umpy
.
transpose
(
kernel
.
value
,
(
3
,
2
,
0
,
1
))
self
.
params
[
kernel_name
]
=
n
p
.
transpose
(
kernel
.
value
,
(
3
,
2
,
0
,
1
))
input_name
=
input
.
name
if
data_format
==
"NHWC"
:
...
...
@@ -1591,7 +1591,7 @@ class TFOpMapper():
kernel
=
"paddle.reshape"
,
inputs
=
{
"x"
:
transpose_name
},
outputs
=
[
reshape_name
],
shape
=
shape
)
shape
=
list
(
shape
)
)
#crop
attrs
=
{}
crop_shape
=
shape
...
...
@@ -1628,7 +1628,8 @@ class TFOpMapper():
inputs
=
{
"x"
:
input_name
},
outputs
=
[
pad_name
],
pad
=
paddings
,
value
=
constant_values
)
value
=
constant_values
,
data_format
=
string
(
'NHWC'
))
#reshape
n
,
h
,
w
,
c
=
x
.
out_shapes
[
0
]
h
=
h
+
paddings
[
2
]
+
paddings
[
3
]
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录