提交 7a8eff36 编写于 作者: M Michal Gallus

Fix old FC backward weights descriptor creation

test=develop
上级 ad61e1b2
...@@ -282,7 +282,7 @@ class FCMKLDNNGradOpKernel : public paddle::framework::OpKernel<T> { ...@@ -282,7 +282,7 @@ class FCMKLDNNGradOpKernel : public paddle::framework::OpKernel<T> {
? mkldnn::inner_product_backward_weights::desc( ? mkldnn::inner_product_backward_weights::desc(
src, diff_weights, bias, diff_dst) src, diff_weights, bias, diff_dst)
: mkldnn::inner_product_backward_weights::desc( : mkldnn::inner_product_backward_weights::desc(
src, diff_weights, bias, diff_dst); src, diff_weights, diff_dst);
return mkldnn::inner_product_backward_weights::primitive_desc( return mkldnn::inner_product_backward_weights::primitive_desc(
bwd_weight_desc, engine, pd); bwd_weight_desc, engine, pd);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册