Eliding casts

[_sean_silva]

EDIT: ignore this post.

@mehdi_amini

Seems like a canonicalization on the extract_element maybe?

That is of limited applicability. Consider if there are two extract_element ops using the same tensor (or another random use by an operator that has the same property that extract_element does).