Support more control flow ops. Delete ops that require ref inputs (e.g.

variables); these ops are not applicable to layout optimizer, because
layout-agnostic nodes are never directly connected to variables, but to
other layout-sensitive nodes such as conv and pooling.

PiperOrigin-RevId: 179699124