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() 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> 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
-
~RMMTensor() override = default