提交 c8cb78fb 编写于 作者: 0 0x45f

Polish code

上级 272dabd7
...@@ -75,7 +75,7 @@ static void clear_unused_out_var_in_backward( ...@@ -75,7 +75,7 @@ static void clear_unused_out_var_in_backward(
delete garbages; delete garbages;
} }
static std::vector<paddle::Tensor> filte_unused_input_var_in_backward( static std::vector<paddle::Tensor> filter_unused_input_var_in_backward(
const std::vector<paddle::Tensor>& x, const std::vector<paddle::Tensor>& x,
const paddle::framework::BlockDesc* backward_block) { const paddle::framework::BlockDesc* backward_block) {
auto filter_x = std::vector<paddle::Tensor>(x); auto filter_x = std::vector<paddle::Tensor>(x);
...@@ -130,7 +130,7 @@ inline void run_program_ad_func( ...@@ -130,7 +130,7 @@ inline void run_program_ad_func(
paddle::framework::BlockDesc*, attrs.at("backward_global_block")); paddle::framework::BlockDesc*, attrs.at("backward_global_block"));
// Clear unused x vars // Clear unused x vars
auto filter_x = auto filter_x =
filte_unused_input_var_in_backward(x, backward_global_block); filter_unused_input_var_in_backward(x, backward_global_block);
grad_node->SetFwdX(filter_x); grad_node->SetFwdX(filter_x);
// Clear unused out vars // Clear unused out vars
clear_unused_out_var_in_backward(out, backward_global_block, step_scope[0]); clear_unused_out_var_in_backward(out, backward_global_block, step_scope[0]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册