(Latest Version)

Class RMMTensor

class RMMTensor : public morpheus::ITensor

TODO(Documentation)

Public Functions

RMMTensor(std::shared_ptr<rmm::device_buffer> device_buffer, TensorIndex 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 TensorIndex bytes() const final

TODO(Documentation)

virtual TensorIndex 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)

© Copyright 2023, NVIDIA. Last updated on Apr 11, 2023.