Rewrite tfl.batch_matmul with batch_1 constants of ones into reduce_sum, to avoid materializing the constant.
Rewrite tfl.batch_matmul with batch_1 constants of ones into reduce_sum, to avoid materializing the constant.
