Attributes used in layout_transform operator. More...

#include <manipulate.h>

Inheritance diagram for tvm::relax::LayoutTransformAttrs:

Collaboration diagram for tvm::relax::LayoutTransformAttrs:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.LayoutTransformAttrs", LayoutTransformAttrs, BaseAttrsNode)

Public Member Functions inherited from tvm::AttrsNodeReflAdapter< LayoutTransformAttrs >
void	InitByPackedArgs (const ffi::PackedArgs &args, bool allow_unknown) final
	Initialize the attributes by arguments. More...

Public Member Functions inherited from tvm::BaseAttrsNode
virtual	~BaseAttrsNode ()
	virtual destructor More...

template<typename... Args>
void	InitBySeq (Args &&... args)
	Initialize the attributes by sequence of arguments. More...

	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", BaseAttrsNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
tir::IndexMap	index_map

ffi::Optional< PrimValue >	pad_value

ffi::Optional< ffi::Array< IntImm > >	axis_separators

ffi::Optional< ffi::Array< IntImm > >	input_axis_separators

Additional Inherited Members
Static Public Attributes inherited from tvm::BaseAttrsNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

Detailed Description

Attributes used in layout_transform operator.

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::LayoutTransformAttrs::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::LayoutTransformAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"relax.attrs.LayoutTransformAttrs"	,
		LayoutTransformAttrs	,
		BaseAttrsNode
	)

Member Data Documentation

◆ axis_separators

ffi::Optional<ffi::Array<IntImm> > tvm::relax::LayoutTransformAttrs::axis_separators

axis_separators between input axes when generating flattened output axes. For buffers representing flat 1-d memory (e.g. any buffer in RAM), this should be an empty array. For buffers representing non-flat memory, each entry in axis_separators should be the first input axis that is part of a new flattened axis.

◆ index_map

tir::IndexMap tvm::relax::LayoutTransformAttrs::index_map

◆ input_axis_separators

ffi::Optional<ffi::Array<IntImm> > tvm::relax::LayoutTransformAttrs::input_axis_separators

axis_separators for input buffers. Needed to identify if the input buffer to layout_transform contains axis separator.

◆ pad_value

ffi::Optional<PrimValue> tvm::relax::LayoutTransformAttrs::pad_value

The documentation for this struct was generated from the following file:

include/tvm/relax/attrs/manipulate.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members