CoTinker wrote: I don't know if changing the type of padding to `2DTensorOf<[Tosa_Int32Or64]>` is a better choice. https://github.com/llvm/llvm-project/blob/902b2a26ab9e1e78dfb66b52fba4512c91472e09/mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td#L1581-L1586 https://github.com/llvm/llvm-project/pull/106351