"$1.getType().cast<ShapedType>())">>;

// Returns the permutation of the trivial reshape op, this will be used to
// construct the transpose op.
def GetPermutationFromTrivialReshape : NativeCodeCall<
  "TFL::GetPermutationFromTrivialReshape("
    "$0.getType().cast<ShapedType>(),"
    "$1.getType().cast<ShapedType>())">;

// Constraint that checks if all values in offset between two
// attributes are non-negative.

      (TF_TransposeOp:$transpose_out1 $input, (Arith_ConstantOp:$permutation1 $p1)), $_),
  (TF_TransposeOp:$transpose_op $input,
      (Arith_ConstantOp
        (RemapPermutation $permutation1,
            (GetPermutationFromTrivialReshape $transpose_out1, $output)))),
  [(IsReshapeEquivalentToTranspose $transpose_out1, $output),
   (AnyStaticShapeTensor $input),
   (AnyStaticShapeTensor $output),
   (HasOneUse $transpose_out1)]>;

}

// Returns the permutation that maps the input shape to the output shape.
// This is only valid for trivial reshape ops.
inline DenseElementsAttr GetPermutationFromTrivialReshape(
    ShapedType input_type, ShapedType output_type) {
  ArrayRef<int64_t> in_shape = input_type.getShape();
  ArrayRef<int64_t> out_shape = output_type.getShape();