column_num: the number of columns; row_num: the number of rows; k_rep: the number of input matrixs that are processed by one graph call. To access more details, see AIE APIs Overview.