$act_fn),
    (TFL_ReshapeOp (BinaryOp $input1, $input2, $act_fn), $shape1),
    [(IsTailOfShape $rhs, $lhs),
      (IsTailOfShape $lhs, $rhs),
      (IsTailOfShape $input1, $input2),
      (IsTailOfShape $input2, $input1),
      (SameElementType $input1, $input2)]>;

  // Move binary op batched RHS before reshape:

}

// Returns whether if `type1` dimensions are the same as the ending dimensions
// of `type2`. This is more restricted than broadcastable.
bool IsTailOfShape(Type type1, Type type2) {
  auto tail_type = mlir::dyn_cast<ShapedType>(type1);
  auto full_type = mlir::dyn_cast<ShapedType>(type2);
  if (!tail_type || !full_type || !tail_type.hasRank() ||