NVIDIA Morpheus (24.10.01)

Struct ITensor

Base Types

Derived Type

struct ITensor : public morpheus::ITensorStorage, public morpheus::ITensorOperations

Subclassed by morpheus::RMMTensor

Public Functions

~ITensor() override = default
virtual RankType rank() const = 0
virtual TensorSize count() const = 0
virtual DType dtype() const = 0
virtual TensorIndex shape(TensorIndex) const = 0
virtual TensorIndex stride(TensorIndex) const = 0
virtual intptr_t stream() const = 0
virtual bool is_compact() const = 0
inline ShapeType get_shape() const
inline ShapeType get_stride() const

Previous Struct InferenceMemoryNLPInterfaceProxy
Next Struct ITensorOperations
© Copyright 2024, NVIDIA. Last updated on Dec 3, 2024.