Class RMMTensor#

Defined in File rmm_tensor.hpp

Inheritance Relationships#

Base Type#

public morpheus::ITensor (Struct ITensor)

Class Documentation#

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)