updated extender (#7763)

This commit is contained in:
Yegor Kruglov 2021-10-01 13:03:49 +03:00 committed by GitHub
parent 7fd0b9be48
commit f3be68d164
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -25,6 +25,9 @@ class GroupConvolutionBackpropData_extender(Extender):
def common_backpropdata_extender(op: Node): def common_backpropdata_extender(op: Node):
for attr in ['strides', 'output_padding', 'pads_begin', 'pads_end', 'dilations']:
Extender.attr_to_list(op, attr)
if op.has_valid('output_padding'): if op.has_valid('output_padding'):
op.output_padding = int64_array([0, 0] + op.output_padding) op.output_padding = int64_array([0, 0] + op.output_padding)