Fix padding attribute error.

enforce_failed
wanghaoshuang 8 years ago
parent 9f8e498138
commit 7c30251d16

@ -83,8 +83,8 @@ void PadGradFunction(const framework::ExecutionContext& context) {
context.op().GetAttr<std::vector<std::pair<int, int>>>("paddings");
Eigen::array<std::pair<int, int>, D> paddings;
for (int i = 0; i < pads.size(); ++i) {
paddings[0].first = -paddings[0].first;
paddings[1].second = -paddings[1].second;
paddings[i].first = -pads[i].first;
paddings[i].second = -pads[i].second;
}
auto* dOut = context.Input<Tensor>(framework::GradVarName("Out"));
auto* dX = context.Output<Tensor>(framework::GradVarName("X"));

Loading…
Cancel
Save