auto arg0 = ops::_Arg(scope.WithOpName("a_0_arg"), DT_INT32, 0);
  auto arg1 = ops::_Arg(scope.WithOpName("b_0_arg"), DT_FLOAT, 1);
  auto arg2 = ops::_Arg(scope.WithOpName("c_0_arg"), DT_INT32, 2);
  auto arg3 = ops::_Arg(scope.WithOpName("d_0_arg"), DT_FLOAT, 3);

  auto arg4 = ops::_Arg(scope.WithOpName("u_0_arg"), DT_RESOURCE, 4);
  auto arg5 = ops::_Arg(scope.WithOpName("v_0_arg"), DT_RESOURCE, 5);

      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);
    }