Class RMMTensor

Base Type

class RMMTensor : public morpheus::ITensor

TODO(Documentation)

Public Functions

RMMTensor(std::shared_ptr<rmm::device_buffer> device_buffer, size_t offset, DType dtype, std::vector<TensorIndex> shape, std::vector<TensorIndex> stride = {})
~RMMTensor() override = default
virtual bool is_compact() const final

TODO(Documentation)

virtual DataType 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 std::vector<TensorIndex> &dims) const override

TODO(Documentation)

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

TODO(Documentation)

virtual std::shared_ptr<ITensor> copy_rows(const std::vector<std::pair<TensorIndex, TensorIndex>> &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 std::size_t bytes() const final

TODO(Documentation)

virtual std::size_t count() const final

TODO(Documentation)

virtual std::size_t shape(std::size_t idx) const final

TODO(Documentation)

virtual std::size_t stride(std::size_t idx) const final

TODO(Documentation)

virtual void *data() const override

TODO(Documentation)

void get_shape(std::vector<TensorIndex> &s) const

TODO(Documentation)

void get_stride(std::vector<TensorIndex> &s) const

TODO(Documentation)

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

TODO(Documentation)

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.