0209a849ef4d520367c2fa880e5605d03106d22d,python/test_scripts/gen_fc.py,,gen_test,#,6

Before Change

    w = tf.constant(tf.random.uniform([512,256])).numpy()
    b = tf.constant(tf.random.uniform([512,1])).numpy()
    // Combine ops to behave like final kernel
    out_1 = tf.math.add(tf.linalg.matmul(w, tf.reshape(in1, (-1,1))), b).numpy()

    in_ref_name = "s_ref_in_%d" % test_number
    w_ref_name = "s_ref_w_%d" % test_number

After Change

    in0 = tf.constant(tf.random.uniform([1,2,2,64])).numpy()
    w = tf.constant(tf.random.uniform([512,256])).numpy()
    if test_number == 0:
        b = np.zeros([1,512], dtype=np.float32).flatten()
    else:
        b = tf.constant(tf.random.uniform([1,512])).numpy().flatten()
    // Combine ops to behave like final kernel
    in1 = tf.reshape(in0, (1,-1)).numpy()
    m = tf.linalg.matmul(in1, w, transpose_b=True)
    print(m.shape)
    out_1 = tf.math.add(m, b).numpy()
    w = np.transpose(w)

    in_ref_name = "s_ref_in_%d" % test_number

Instances