tf.broadcast_static_shape