CHECK_EQ(output_shapes->num_items, 1);

    int num_dims = output_shapes->items[0].num_dims;
    int64_t* dims = output_shapes->items[0].dims;

    if (!expected_shape.has_value()) {
      EXPECT_EQ(num_dims, -1);
      EXPECT_EQ(dims, nullptr);
      return;
    }

    EXPECT_EQ(num_dims, expected_shape->size());
    for (size_t i = 0; i < num_dims; ++i) {
      EXPECT_EQ(dims[i], (*expected_shape)[i]);
    }

                                 const int64_t* dims, int num_dims) {
  DCHECK(index >= 0 && index < shape_list->num_items);
  TF_ShapeAndType& shape = shape_list->items[index];
  DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!";
  DCHECK(num_dims >= 0) << "Number of dimensions cannot be negative!";
  shape.num_dims = num_dims;
  shape.dims = new int64_t[num_dims];
  memcpy(shape.dims, dims, sizeof(int64_t) * num_dims);

// Read the shape of a variable and write to `dims`
TF_CAPI_EXPORT extern void TF_CheckpointReaderGetVariableShape(
    TF_CheckpointReader* reader, const char* name, int64_t* dims, int num_dims,
    TF_Status* status);
// Get the number of dimension of a variable
TF_CAPI_EXPORT extern int TF_CheckpointReaderGetVariableNumDims(
    TF_CheckpointReader* reader, const char* name);
// Load the weight of a variable

        lsarpc.LsarSidArray decodedArray = new lsarpc.LsarSidArray();
        decodedArray.decode(decodeBuffer);

        // Verify the decoded data matches
        assertEquals(sidArray.num_sids, decodedArray.num_sids);
        assertNotNull(decodedArray.sids);
        assertEquals(2, decodedArray.sids.length);

        // Verify first SID
        assertNotNull(decodedArray.sids[0]);

        }

        /** Number of SIDs in the array. */
        public int num_sids;
        /** Array of SID pointers. */
        public LsarSidPtr[] sids;

        @Override
        public void encode(NdrBuffer _dst) throws NdrException {
            _dst.align(4);
            _dst.enc_ndr_long(this.num_sids);
            _dst.enc_ndr_referent(this.sids, 1);

            if (this.sids != null) {

        }

        /** Number of SIDs in the array. */
        public int num_sids;
        /** Array of SID pointers. */
        public LsarSidPtr[] sids;

        @Override
        public void encode(NdrBuffer _dst) throws NdrException {
            _dst.align(4);
            _dst.enc_ndr_long(num_sids);
            _dst.enc_ndr_referent(sids, 1);

            if (sids != null) {

                    if (rpc.retval != 0) {
                        throw new SmbException(rpc.retval, false);
                    }
                    final SID[] sids = new SID[rpc.sids.num_sids];

                    final String origin_server = handle.getServer();
                    final CIFSContext origin_ctx = handle.getTransportContext();

                    for (int i = 0; i < sids.length; i++) {

            handle.sendrecv(rpc);
            if (rpc.retval != 0) {
                throw new SmbException(rpc.retval, false);
            }
            final SID[] sids = new SID[rpc.sids.num_sids];

            final String origin_server = handle.getServer();
            final NtlmPasswordAuthentication origin_auth = (NtlmPasswordAuthentication) handle.getPrincipal();