for (const auto& entry : resource_arg_to_new_output) {
      int64_t resource_arg_index = entry.getFirst();
      int64_t output_index = entry.getSecond();
      new_retvals[output_index] = branch.getArgument(resource_arg_index);
    }
    auto old_return = branch.front().getTerminator();
    OpBuilder builder(old_return);
    auto new_return =

          /*axis=*/0);
      for (auto index_item : llvm::zip(index_attr, items.getResults())) {
        int64_t output_index = std::get<0>(index_item).getSExtValue();
        Value item = std::get<1>(index_item);
        values[output_index] = item;
      }
    }

    rewriter.replaceOpWithNewOp<PackOp>(op, op.getType(), values);
    return success();
  }
};

              Eq(-128));
  const int output_idx = 4;
  EXPECT_THAT(subgraph->tensors[output_idx]->type, Eq(TensorType_INT8));
  EXPECT_THAT(subgraph->tensors[output_idx]->name, Eq("output"));
  EXPECT_THAT(subgraph->tensors[output_idx]->quantization->scale, SizeIs(1));
  EXPECT_THAT(subgraph->tensors[output_idx]->quantization->zero_point,
              SizeIs(1));