Fix av1_write_coeffs_mb when ext_partition is on

Use av1_foreach_transformed_block_in_plane.

Change-Id: I22f92b601f6a90e92a93dbff8492df86c717ab6f
diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c
index b597ec7..1c04554 100644
--- a/av1/encoder/bitstream.c
+++ b/av1/encoder/bitstream.c
@@ -2097,7 +2097,7 @@
 #endif  // CONFIG_RD_DEBUG
       } else {
 #if CONFIG_LV_MAP
-        av1_write_coeffs_mb(cm, x, w, plane);
+        av1_write_coeffs_mb(cm, x, w, plane, bsize);
 #else
         const TX_SIZE tx = av1_get_tx_size(plane, xd);
         const int bkw = tx_size_wide_unit[tx];