Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
溜达小灰鼠
Paddle
提交
fbc30215
P
Paddle
项目概览
溜达小灰鼠
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
fbc30215
编写于
1月 09, 2018
作者:
F
fengjiayi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refine WhileGradOp code
上级
8f962f74
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
7 addition
and
17 deletion
+7
-17
paddle/operators/while_op.cc
paddle/operators/while_op.cc
+7
-17
未找到文件。
paddle/operators/while_op.cc
浏览文件 @
fbc30215
...
...
@@ -219,18 +219,15 @@ class WhileGradOpDescMaker : public framework::SingleGradOpDescMaker {
auto
*
grad_block
=
this
->
grad_block_
[
0
];
auto
*
fwd_block
=
grad_block
->
ParentBlock
();
// auto *parent_block = fwd_block->ParentBlock();
// Not all of IGs will be generated by inner gradient operators of while op.
// Ignore IGs that is not generated by the inside block.
std
::
unordered_set
<
std
::
string
>
inner_op_outputs
;
LOG
(
INFO
)
<<
"FUCK1"
;
for
(
const
auto
*
op
:
grad_block
->
AllOps
())
{
for
(
auto
&
oname
:
op
->
OutputArgumentNames
())
{
inner_op_outputs
.
insert
(
oname
);
}
}
LOG
(
INFO
)
<<
"FUCK2"
;
auto
igs
=
InputGrad
(
kX
,
/*do not drop empty gradient*/
false
);
for
(
auto
&
each_ig
:
igs
)
{
if
(
inner_op_outputs
.
find
(
each_ig
)
==
inner_op_outputs
.
end
())
{
...
...
@@ -243,11 +240,13 @@ class WhileGradOpDescMaker : public framework::SingleGradOpDescMaker {
// OG should be re-calculated by step blocks, since many outputs of while op
// do not need to calculate gradients.
std
::
unordered_set
<
std
::
string
>
block_ins
;
std
::
copy
(
Input
(
kX
).
begin
(),
Input
(
kX
).
end
(),
std
::
inserter
(
block_ins
,
block_ins
.
end
()));
std
::
copy
(
Output
(
kOutputs
).
begin
(),
Output
(
kOutputs
).
end
(),
std
::
inserter
(
block_ins
,
block_ins
.
end
()));
block_ins
.
reserve
(
Input
(
kX
).
size
()
+
Output
(
kOutputs
).
size
());
for
(
auto
&
p
:
Input
(
kX
))
{
block_ins
.
insert
(
p
);
}
for
(
auto
&
o
:
Output
(
kOutputs
))
{
block_ins
.
insert
(
o
);
}
std
::
unordered_set
<
std
::
string
>
extra_inputs
;
for
(
const
auto
*
op
:
grad_block
->
AllOps
())
{
for
(
auto
&
input_name
:
op
->
InputArgumentNames
())
{
...
...
@@ -257,15 +256,6 @@ class WhileGradOpDescMaker : public framework::SingleGradOpDescMaker {
fwd_block
->
FindVar
(
input_name
)
!=
nullptr
)
{
continue
;
}
/*
if (parent_block->FindVarRecursive(input_name) == nullptr) {
VLOG(5) << "WARNING! Variable '" << input_name
<< "' is the input of '" << op->Type()
<< "'. But can not be found in any block.";
continue;
}
*/
extra_inputs
.
insert
(
input_name
);
}
for
(
auto
&
output_name
:
op
->
OutputArgumentNames
())
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录