24 #ifndef TVM_RELAX_ATTRS_LINEAR_ALGEBRA_H_
25 #define TVM_RELAX_ATTRS_LINEAR_ALGEBRA_H_
39 "The data type of the output tensor");
42 static constexpr
const char*
_type_key =
"relax.attrs.MatmulAttrs";
53 "The einsum expression string");
56 static constexpr
const char*
_type_key =
"relax.attrs.EinsumAttrs";
Adapter for AttrsNode with the new reflection API.
Definition: attrs.h:384
Base class of all attribute class.
Definition: attrs.h:103
Runtime primitive data type.
Definition: data_type.h:47
Definition: repr_printer.h:91
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Attributes used in einsum operator.
Definition: linear_algebra.h:47
static void RegisterReflection()
Definition: linear_algebra.h:50
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(EinsumAttrs, BaseAttrsNode)
static constexpr const char * _type_key
Definition: linear_algebra.h:56
String subscripts
Definition: linear_algebra.h:48
Attributes for matmul operator.
Definition: linear_algebra.h:33
static void RegisterReflection()
Definition: linear_algebra.h:36
TVM_FFI_DECLARE_FINAL_OBJECT_INFO(MatmulAttrs, BaseAttrsNode)
static constexpr const char * _type_key
Definition: linear_algebra.h:42
DataType out_dtype
Definition: linear_algebra.h:34