Update python fprop to use GEMM kernels.