def METHOD_NAME(weights_a, weights_b, equal=True):
    for i, _ in enumerate(weights_a):
        if equal:
            assert tf.reduce_all(tf.equal(weights_a[i], weights_b[i]))
        else:
            assert tf.reduce_all(tf.not_equal(weights_a[i], weights_b[i]))