Method AttachBuffer

dali/c_api_2/data_objects.h:437–540

  }

  void AttachBuffer(
      int num_samples,
      int ndim,
      const int64_t *shapes,
      daliDataType_t dtype,
      const char *layout,
      void *data,
      const ptrdiff_t *sample_offsets,
      daliDeleter_t deleter) override {
    ValidateShape(num_samples, ndim, shapes);
    Validate(dtype);

    if (!data && num_samples > 0) {
      for (int i = 0; i < num_samples; i++) {
        auto sample_shape = make_cspan(&shapes[i*ndim], ndim);

        if (volume(sample_shape) > 0)
          throw std::invalid_argument(
            "The pointer to the data buffer must not be null for a non-empty tensor list.");

        if (sample_offsets && sample_offsets[i])
          throw std::invalid_argument(
            "All sample_offsets must be zero when the data pointer is NULL.");
      }
    }

    TensorLayout new_layout = {};

    if (!layout) {
      if (ndim == tl_->sample_dim())
        new_layout = tl_->GetLayout();
    } else {
      new_layout = layout;
      Validate(new_layout, ndim);
    }

    tl_->Reset();
    tl_->SetSize(num_samples);
    tl_->set_sample_dim(ndim);
    tl_->SetLayout(new_layout);
    tl_->set_type(dtype);
    ptrdiff_t next_offset = 0;
    auto type_info = TypeTable::GetTypeInfo(dtype);
    auto element_size = type_info.size();

    std::shared_ptr<void> buffer;
    if (!deleter.delete_buffer && !deleter.destroy_context) {
      buffer = std::shared_ptr<void>(data, [](void *){});
    } else {
      buffer = std::shared_ptr<void>(data, BufferDeleter{deleter, tl_->order()});
    }

    bool is_contiguous = true;
    if (sample_offsets) {
      for (int i = 0; i < num_samples; i++) {
        if (sample_offsets[i] != next_offset) {
          is_contiguous = false;
          break;