}
}

}  // namespace

void TransposeOp::getCanonicalizationPatterns(RewritePatternSet& results,
                                              MLIRContext* context) {
  results.add<ConvertTransposeToDecreaseRank>(context);
}

OpFoldResult TransposeOp::fold(FoldAdaptor adaptor) {
  auto operands = adaptor.getOperands();

  auto input_tensor = operands[0].dyn_cast_or_null<DenseElementsAttr>();