Add extended partition size in txfm_partition_context()

Change-Id: I66fdc495b337debbc8599c9b9ae124cb27758b92
diff --git a/av1/common/onyxc_int.h b/av1/common/onyxc_int.h
index 7079cdd..3b33351 100644
--- a/av1/common/onyxc_int.h
+++ b/av1/common/onyxc_int.h
@@ -924,6 +924,9 @@
   if (tx_size <= TX_4X4) return 0;
 
   switch (AOMMAX(block_size_wide[bsize], block_size_high[bsize])) {
+#if CONFIG_EXT_PARTITION
+    case 128:
+#endif
     case 64:
     case 32: max_tx_size = TX_32X32; break;
     case 16: max_tx_size = TX_16X16; break;