Template Struct arg#

Inheritance Relationships#

Base Type#

Struct Documentation#

template<typename T = Expr>
struct arg : public tvm::sig::SignatureTrait#

One value argument of type T, for example sig::arg<PrimExpr>("index") for a required primitive index expression.

Public Functions

inline explicit arg(ffi::String name, ffi::String doc = "")#
inline void CollectMetadata(OpObj *op) const#

Public Members

ffi::String name#

Argument name recorded in operator metadata.

ffi::String doc#

Argument documentation recorded in operator metadata.

Public Static Functions

static inline bool Validate(ValidateState *state, ffi::Expected<void> *out)#

Public Static Attributes

static constexpr size_t kArgsCount = 1#

Contributes one required value argument.