if (it.index() != it.value()) return {};
  }

  if (op.getType() != op.getX().getType()) return {};
  return op.getX();
}

OpFoldResult FoldCancellableTranspose(TransposeOp op) {
  // Operand is a TransposeOp.
  auto transpose = dyn_cast_or_null<TF::TransposeOp>(op.getX().getDefiningOp());
  if (!transpose) return {};