Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
未来在这儿
X2Paddle
提交
24e799a9
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,体验更适合开发者的 AI 搜索 >>
提交
24e799a9
编写于
8月 29, 2019
作者:
J
jiangjiajun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add more models support for tensorflow
上级
7ca9c323
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
0 addition
and
25 deletion
+0
-25
x2paddle/op_mapper/tf_op_mapper.py
x2paddle/op_mapper/tf_op_mapper.py
+0
-6
x2paddle/optimizer/tf_optimizer.py
x2paddle/optimizer/tf_optimizer.py
+0
-19
未找到文件。
x2paddle/op_mapper/tf_op_mapper.py
浏览文件 @
24e799a9
...
...
@@ -314,15 +314,9 @@ class TFOpMapper(OpMapper):
input_name
=
input_name
+
"[{}]"
.
format
(
input
.
index
)
node
.
fluid_code
.
add_layer
(
"{} = {}"
).
format
(
node
.
layer_name
,
input_name
)
#
# node.fluid_code.add_layer("assign",
# inputs=input,
# output=node,
# param_attr=None)
node
.
tf_data_format
=
"NHWC"
self
.
graph
.
data_format_propagation
(
node
)
elif
len
(
input
.
out_shapes
[
0
])
>
4
:
print
(
input
.
layer_name
,
input
.
tf_data_format
,
input
.
pd_data_format
)
tf_data_format
=
list
(
input
.
tf_data_format
)
pd_data_format
=
list
(
input
.
pd_data_format
)
new_perm
=
[
i
for
i
in
range
(
len
(
perm
))]
...
...
x2paddle/optimizer/tf_optimizer.py
浏览文件 @
24e799a9
...
...
@@ -39,11 +39,6 @@ class TFOptimizer(object):
self
.
graph
=
op_mapper
.
graph
def
delete_redundance_code
(
self
):
# print("==========omit_nodes============")
# for node_name in set(self.op_mapper.omit_nodes):
# node = self.graph.get_node(node_name)
# print(node.layer_name, self.op_mapper.omit_nodes.count(node.layer_name), len(node.outputs), node.outputs)
# print("================================")
for
node_name
in
self
.
graph
.
topo_sort
:
if
node_name
in
self
.
op_mapper
.
omit_nodes
:
node
=
self
.
graph
.
get_node
(
node_name
)
...
...
@@ -67,13 +62,6 @@ class TFOptimizer(object):
del
self
.
graph
.
node_map
[
node_name
]
def
strip_graph
(
self
):
# print("=============")
# for i, node_name in enumerate(self.graph.topo_sort):
# node = self.graph.get_node(node_name)
# if node is None:
# continue
# print(node.layer_name, node.inputs)
# print("================")
visited_nodes
=
set
()
def
visit
(
node_name
):
...
...
@@ -87,10 +75,6 @@ class TFOptimizer(object):
for
node_name
in
self
.
graph
.
output_nodes
:
visit
(
node_name
)
# print("=============visited nodes++++++++++++")
# for name in visited_nodes:
# print(name)
# print("===================================")
for
i
,
node_name
in
enumerate
(
self
.
graph
.
topo_sort
):
if
node_name
not
in
visited_nodes
:
node
=
self
.
graph
.
get_node
(
node_name
)
...
...
@@ -221,9 +205,6 @@ class TFOptimizer(object):
if
out_node
.
layer_type
==
"BiasAdd"
:
del
out_node
.
fluid_code
.
layers
[
0
]
out_node
.
fluid_code
.
layers
[
0
].
inputs
[
'x'
]
=
last_out
# out_node.fluid_code.layers[0].param_attr["axis"] = 1
else
:
del
out_node
.
fluid_code
.
layers
[
0
]
out_node
.
fluid_code
.
layers
[
0
].
inputs
=
last_out
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录