Class RMMTensor

• Defined in File rmm_tensor.hpp

Base Type

• public morpheus::ITensor (Struct ITensor)

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)