// Constraint that checks if the transpose op is trivial. Trivial means that
// the permutation is a cyclic permutation of the original shape with only the
// identity dimensions permuted.
def IsTransposeTrivial : Constraint<CPred<
  "TFL::IsTransposeTrivial($0.getType().cast<ShapedType>().getShape(), $1)">>;

// Constraint that checks if the reshape op is equivalent to a transpose op.

}

// Returns true if the transpose op is trivial. Trivial means that
// the permutation is a cyclic permutation of the original shape with only the
// identity dimensions permuted.
inline bool IsTransposeTrivial(llvm::ArrayRef<int64_t> input_shape,
                               Value perm) {
  DenseElementsAttr perm_values_attr;
  if (!matchPattern(perm, m_Constant(&perm_values_attr))) return false;

def ConvertTrivialTransposeOpToReshapeOp : Pat<
  (TFL_TransposeOp:$transpose_op $input, (Arith_ConstantOp:$permutation $p1)),
  (TFL_ReshapeOp $input, (Arith_ConstantOp (GetShape $transpose_op))),
  [(IsTransposeTrivial $input, $permutation),
   (AnyStaticShapeTensor $input),
   (AnyStaticShapeTensor $transpose_op)]>;

// Pattern to fuse redundant tanspose op
def FoldDoubleTranspose : Pat<
  (TFL_TransposeOp