// Enable an optimization pass that transforms FC to BatchMatmul only when
    // `unfold_batch_matmul=false`.
    pass_manager.addNestedPass<mlir::func::FuncOp>(
        mlir::TFL::CreateOptimizeBatchMatmulPass());
  }
  pass_manager.addNestedPass<mlir::func::FuncOp>(
      mlir::TFL::CreateOptimizePass(/*enable_canonicalization=*/true));
}