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 Apr 25, 2024.