issue with tf broadcasting unresolved