A couple of extensions to rewriter #8444

Job	Run time
update_readme	5s
build_docs (ubuntu-latest)	13s
test (ubuntu-latest, py312)	19s
build_docs (windows-latest)	18s
test (ubuntu-latest, py311)	21s
test (ubuntu-latest, py311-torch-nightly)	18s
test (ubuntu-latest, py311-onnx-weekly)	19s
test (ubuntu-latest, py311-ort-nightly)	17s
test (ubuntu-latest, py310)	21s
test (windows-latest, py312)	18s
test (windows-latest, py311)	17s
test (windows-latest, py311-torch-nightly)	18s
test (windows-latest, py311-onnx-weekly)	21s
test (windows-latest, py311-ort-nightly)	20s
test (windows-latest, py310)	21s
test (macos-latest, py312)	37s
test (macos-latest, py311)	31s
test (macos-latest, py311-torch-nightly)	31s
test (macos-latest, py311-onnx-weekly)	35s
test (macos-latest, py311-ort-nightly)	34s
test (macos-latest, py310)	16s
	7m 30s

Provide feedback