tvm
tvm::tirx::TileLayoutNode Member List

This is the complete list of members for tvm::tirx::TileLayoutNode, including all inherited members.

_type_s_eq_hash_kindtvm::tirx::LayoutNodestatic
Apply(ffi::Array< PrimExpr > coord) const finaltvm::tirx::TileLayoutNodevirtual
Apply(PrimExpr coord) const finaltvm::tirx::TileLayoutNodevirtual
tvm::tirx::LayoutNode::Apply(const ffi::Array< PrimExpr > &coord, const ffi::Array< PrimExpr > &shape) consttvm::tirx::LayoutNode
Canonicalize() const finaltvm::tirx::TileLayoutNodevirtual
CompatibleWithShape(const ffi::Array< PrimExpr > &shape) const finaltvm::tirx::TileLayoutNodevirtual
DefaultLayout(ffi::Array< PrimExpr > shape)tvm::tirx::TileLayoutNodestatic
DirectSum(const TileLayout &left, const ffi::Array< PrimExpr > &left_shape, const ffi::Array< PrimExpr > &right_shape) const finaltvm::tirx::TileLayoutNodevirtual
GetScope() consttvm::tirx::TileLayoutNode
GetShardShape() consttvm::tirx::TileLayoutNode
GetSize(ffi::Optional< ffi::String > axis_name=std::nullopt) const finaltvm::tirx::TileLayoutNodevirtual
GetSpan(ffi::Optional< ffi::String > axis_name=std::nullopt) const finaltvm::tirx::TileLayoutNodevirtual
HasMemoryAxis() consttvm::tirx::TileLayoutNode
HasThreadAxis() consttvm::tirx::TileLayoutNode
IsDirectSumLeft(const Layout &sum_layout, const ffi::Array< PrimExpr > &interleaved_shape, const ffi::Array< PrimExpr > &left_shape) const finaltvm::tirx::TileLayoutNodevirtual
IsDirectSumRight(const Layout &sum_layout, const ffi::Array< PrimExpr > &interleaved_shape, const ffi::Array< PrimExpr > &right_shape) const finaltvm::tirx::TileLayoutNodevirtual
IsTileInner(const Layout &tile_layout, const ffi::Array< PrimExpr > &tiled_shape, const ffi::Array< PrimExpr > &inner_shape) const finaltvm::tirx::TileLayoutNodevirtual
IsTileOuter(const Layout &tile_layout, const ffi::Array< PrimExpr > &tiled_shape, const ffi::Array< PrimExpr > &outer_shape) const finaltvm::tirx::TileLayoutNodevirtual
IsTrainium() consttvm::tirx::TileLayoutNode
IsTrivial() consttvm::tirx::TileLayoutNode
offsettvm::tirx::TileLayoutNode
RegisterReflection()tvm::tirx::TileLayoutNodeinlinestatic
replicatvm::tirx::TileLayoutNode
shardtvm::tirx::TileLayoutNode
Slice(const ffi::Array< PrimExpr > &shape, const Region &region) const finaltvm::tirx::TileLayoutNodevirtual
Tile(const TileLayout &outer, const ffi::Array< PrimExpr > &outer_shape, const ffi::Array< PrimExpr > &inner_shape) const finaltvm::tirx::TileLayoutNodevirtual
TVM_FFI_DECLARE_OBJECT_INFO("tirx.Layout", LayoutNode, ffi::Object)tvm::tirx::LayoutNode
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("tirx.TileLayout", TileLayoutNode, LayoutNode)tvm::tirx::TileLayoutNode
VerifyWellFormed() const finaltvm::tirx::TileLayoutNodevirtual