auto tf_dlm_type = GetDlDataType(data_type, status);
  if (!status->status.ok()) {
    return nullptr;
  }

  TensorReference tensor_ref(*tensor);  // This will call buf_->Ref()
  auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);
  tf_dlm_tensor_ctx->reference = tensor_ref;

  DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
  dlm_tensor->manager_ctx = tf_dlm_tensor_ctx;

    ]

    repr_ds: repr_dataset.RepresentativeDataset = [
        {
            'tensor_key': ops.convert_to_tensor(sample),
        }
        for sample in samples
    ]

    # Extend the representative dataset with np.ndarrays.
    repr_ds.extend([
        {'tensor_key': np.random.uniform(low=-1.0, high=1.0, size=(3, 3))}
        for _ in range(4)
    ])

    random.shuffle(repr_ds)