Fix dtype issue in TF BART
Fixes an issue in BART caused by implicit casting of Python ints to tf.int32, when other inputs are tf.int64. We usually get away with this, but tf.where() requires both inputs to have the same dtype.