tvm::tirx::SBijectiveLayoutNode Class Reference

#include <data_layout.h>

Inheritance diagram for tvm::tirx::SBijectiveLayoutNode:

Collaboration diagram for tvm::tirx::SBijectiveLayoutNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.SBijectiveLayout", SBijectiveLayoutNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Array< PrimExpr >	index_forward_rule
	Describes how source axes can be mapped to the destination axes, e.g., [i0 / 16, i1, i0 % 16] can describe NC -> NC16n. More...
ffi::Array< PrimExpr >	index_backward_rule
	Describes how destination axes can be mapped to the source axes. More...
ffi::Array< PrimExpr >	shape_forward_rule
	Describes how source shapes can be mapped to the destination shapes. More...
ffi::Array< PrimExpr >	shape_backward_rule
	Describes how destination shapes can be mapped to the source shapes. More...
SLayout	src_layout
	The source layout. More...
SLayout	dst_layout
	The destination layout. More...

Member Function Documentation

RegisterReflection()

static void tvm::tirx::SBijectiveLayoutNode::RegisterReflection ( )

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tirx::SBijectiveLayoutNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.SBijectiveLayout"	,
		SBijectiveLayoutNode	,
		ffi::Object
	)

Member Data Documentation

dst_layout

SLayout tvm::tirx::SBijectiveLayoutNode::dst_layout

The destination layout.

index_backward_rule

ffi::Array<PrimExpr> tvm::tirx::SBijectiveLayoutNode::index_backward_rule

Describes how destination axes can be mapped to the source axes.

index_forward_rule

ffi::Array<PrimExpr> tvm::tirx::SBijectiveLayoutNode::index_forward_rule

Describes how source axes can be mapped to the destination axes, e.g., [i0 / 16, i1, i0 % 16] can describe NC -> NC16n.

shape_backward_rule

ffi::Array<PrimExpr> tvm::tirx::SBijectiveLayoutNode::shape_backward_rule

Describes how destination shapes can be mapped to the source shapes.

shape_forward_rule

ffi::Array<PrimExpr> tvm::tirx::SBijectiveLayoutNode::shape_forward_rule

Describes how source shapes can be mapped to the destination shapes.

src_layout

SLayout tvm::tirx::SBijectiveLayoutNode::src_layout

The source layout.

---

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

• include/tvm/s_tir/data_layout.h