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