A couple of extensions to rewriter #8438

Job	Run time
update_readme	5s
test (ubuntu-latest, py312)	20s
test (ubuntu-latest, py311)	32s
build_docs (ubuntu-latest)	17s
test (ubuntu-latest, py311-torch-nightly)	21s
build_docs (windows-latest)	15s
test (ubuntu-latest, py311-onnx-weekly)	32s
test (ubuntu-latest, py311-ort-nightly)	24s
test (ubuntu-latest, py310)	21s
test (windows-latest, py312)	11s
test (windows-latest, py311)	23s
test (windows-latest, py311-torch-nightly)	7s
test (windows-latest, py311-onnx-weekly)	24s
test (windows-latest, py311-ort-nightly)	24s
test (windows-latest, py310)	24s
test (macos-latest, py312)	48s
test (macos-latest, py311)	49s
test (macos-latest, py311-torch-nightly)	41s
test (macos-latest, py311-onnx-weekly)	36s
test (macos-latest, py311-ort-nightly)	38s
test (macos-latest, py310)	21s
	8m 53s

Provide feedback