def METHOD_NAME(N, M, banks):
    in1 = np.random.rand(*[banks, N, M]).astype('f')
    in2 = np.random.rand(*[banks, N, M]).astype('f')
    expected = np.sum(in1 * in2, axis=2, dtype=np.float32)
    out = np.zeros((banks, N), dtype=np.float32)
    return (in1, in2, expected, out)