Skip to content

Commit f17c674

Browse files
Andrew Grebenisanfacebook-github-bot
authored andcommitted
Fix op signature for avg_pool2d
Summary: The default for padding was incorrect, adding default with the correct dtype Differential Revision: D83873533
1 parent 54bda26 commit f17c674

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/cadence/aot/ops_registrations.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@
329329
"Tensor bias_scale, float out_scale, int out_zero_point, Tensor out_multiplier, Tensor out_shift, bool channel_last=False) -> (Tensor out)"
330330
)
331331
lib.define(
332-
"avg_pool2d(Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=0, bool ceil_mode=False, "
332+
"avg_pool2d(Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=[], bool ceil_mode=False, "
333333
"bool count_include_pad=True, int? divisor_override=None, Tensor? in_zero_point=None, bool channel_last=False) -> (Tensor out)"
334334
)
335335
lib.define(
@@ -525,7 +525,7 @@
525525
"Tensor out_multiplier, Tensor out_shift, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)"
526526
)
527527
lib.define(
528-
"avg_pool2d.out(Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=0, "
528+
"avg_pool2d.out(Tensor input, int[2] kernel_size, int[2] stride=[], int[2] padding=[], "
529529
"bool ceil_mode=False, bool count_include_pad=True, int? divisor_override=None, "
530530
"Tensor? in_zero_point=None, bool channel_last=False, *, Tensor(a!) out) -> Tensor(a!)"
531531
)

0 commit comments

Comments
 (0)