|
|
|
@@ -1084,7 +1084,7 @@ class Conv2DBackpropInput(PrimitiveWithInfer): |
|
|
|
self.init_prim_io_names(inputs=['out_backprop', 'filter', 'input_sizes'], outputs=['output']) |
|
|
|
self.out_channel = validator.check_integer('out_channel', out_channel, 0, Rel.GT, self.name) |
|
|
|
self.kernel_size = _check_positive_int_or_tuple('kernel_size', kernel_size, self.name) |
|
|
|
self.stride = _check_positive_int_or_tuple('stride', stride, self.name, allow_four=True, ret_four=False) |
|
|
|
self.stride = _check_positive_int_or_tuple('stride', stride, self.name, allow_four=True, ret_four=True) |
|
|
|
self.add_prim_attr('stride', self.stride) |
|
|
|
self.dilation = _check_positive_int_or_tuple('dilation', dilation, self.name, allow_four=True, ret_four=True) |
|
|
|
self.add_prim_attr('dilation', self.dilation) |
|
|
|
|