Method SetNext

dali/pipeline/operator/sequence_shape.h:237–242

  TensorListBuilder(TensorList<Backend> &tv) : tv_{tv} {}  // NOLINT

  void SetNext(const SliceView &view) {
    assert(NextSampleIdx() < tv_.num_samples());
    std::shared_ptr<void> ptr(view.ptr, [](void *) {});  // no deleter
    tv_.SetSample(next_++, ptr, view.type_size * volume(view.shape), tv_.is_pinned(), view.shape,
                  tv_.type(), tv_.device_id(), tv_.order(), tv_.GetLayout());
  }

  int NextSampleIdx() const {
    return next_;