auto in_rtt =
          llvm::dyn_cast<RankedTensorType>(cst_arg->getResult(0).getType());

      auto inverse_type =
          RankedTensorType::get(PermuteShape(in_rtt.getShape(), inverse_perm),
                                in_rtt.getElementType());

      tposed_const = rewriter.create<TFL::TransposeOp>(
          cst_arg->getLoc(), inverse_type, cst_arg->getResult(0),
          inverse_perm_op);
    }