提交 10ba9ab7 authored 作者: Sina Honari's avatar Sina Honari

fixing make_node param passing for conv_grad

上级 c48ee179
......@@ -269,7 +269,7 @@ def conv2d_grad_wrt_inputs(output_grad,
subsample=subsample,
filter_flip=filter_flip)
return grad_input_op(filters, input, output_grad_shape)
return grad_input_op(filters, output_grad, input_shape)
def conv2d_grad_wrt_weights(input,
......@@ -363,7 +363,7 @@ def conv2d_grad_wrt_weights(input,
subsample=subsample,
filter_flip=filter_flip)
return gradWeight_op(input, output_grad, input_shape)
return gradWeight_op(input, output_grad, filter_shape)
class BaseAbstractConv2d(Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论