tree a6b61fdd5189062f607bb90a7d2f28f3ee265582
parent 8d0ae888464fd9e8f509b660c49604b70fcd6306
author A. Unique TensorFlower <gardener@tensorflow.org> 1655149176 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1655149487 -0700

Add sanity check in `tpu_rewrite_pass`.

This is used to early detect IR that violate the assumption in `tpu_rewrite_pass`. Individual resource reads/writes from the unpartitioned resource variable (from `tf.TPUPartitionedInput`) to individual partitioned resource variables (`tf.TPUPartitionedInput` operands) must be present since they are ensured by `TPUResourceReadsWritesPartitioning` pass

PiperOrigin-RevId: 454674859