fixup

Xilinx · Jul 13, 2023 · 832e9b0 · 832e9b0
1 parent 987b43f
commit 832e9b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/quant_tensor/torch_handler.py b/src/brevitas/quant_tensor/torch_handler.py
@@ -157,5 +157,5 @@ def pixel_shuffle_handler(*args, **kwargs):
 
 
 @implements(F.pixel_unshuffle)
-def pixel_shuffle_handler(*args, **kwargs):
+def pixel_unshuffle_handler(*args, **kwargs):
     return quant_invariant_handler(F.pixel_unshuffle, *args, **kwargs)