Add visits to nodes that already have flat_maps in ast::MutVisitor

Add visits to nodes that already have flat_maps in ast::MutVisitor #141537