Class RMMTensor

class RMMTensor : public morpheus::ITensor

TODO(Documentation)

Public Functions

RMMTensor(std::shared_ptr<rmm::device_buffer> device_buffer, TensorSize offset, DType dtype, ShapeType shape, ShapeType stride = {})

~RMMTensor() override = default

virtual bool is_compact() const final

TODO(Documentation)

virtual DType dtype() const override

TODO(Documentation)

virtual RankType rank() const final

TODO(Documentation)

virtual std::shared_ptr<ITensor> deep_copy() const override

TODO(Documentation)

virtual std::shared_ptr<ITensor> reshape(const ShapeType &dims) const override

TODO(Documentation)

virtual std::shared_ptr<ITensor> slice(const ShapeType &min_dims, const ShapeType &max_dims) const override

TODO(Documentation)

virtual std::shared_ptr<ITensor> copy_rows(const std::vector<RangeType> &selected_rows, TensorIndex num_rows) const override

Creates a depp copy of the specified rows specified as vector<pair<start, stop>> not inclusive of the stop row.

Parameters
  • selected_rows

  • num_rows

Returns

std::shared_ptr<ITensor>

virtual std::shared_ptr<MemoryDescriptor> get_memory() const override

TODO(Documentation)

virtual TensorSize bytes() const final

TODO(Documentation)

virtual TensorSize count() const final

TODO(Documentation)

virtual TensorIndex shape(TensorIndex idx) const final

TODO(Documentation)

virtual TensorIndex stride(TensorIndex idx) const final

TODO(Documentation)

virtual void *data() const override

TODO(Documentation)

void get_shape(ShapeType &s) const

TODO(Documentation)

void get_stride(ShapeType &s) const

TODO(Documentation)

virtual intptr_t stream() const override

virtual std::shared_ptr<ITensor> as_type(DType dtype) const override

TODO(Documentation)

Previous Class RESTDataLoader
Next Template Class SerializeStage
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.