Consecutive calls of the the join op ( during the optimization phase) get
the axis as a constant. What used to happen was that in such cases Theano
would not use that constant to define its broadcastable pattern.