MLIR 23.0.0git
mlir::shard::SplitLastAxisPattern Member List

This is the complete list of members for mlir::shard::SplitLastAxisPattern, including all inherited members.

hasStaticOffsets(const Sharding &srcSharding, const Sharding &tgtSharding)mlir::shard::ReshardingPatterninlineprotectedstatic
hasStaticOffsetsOrHalos(const Sharding &srcSharding, const Sharding &tgtSharding)mlir::shard::ReshardingPatterninlineprotectedstatic
tryApply(ImplicitLocOpBuilder &builder, GridOp grid, int64_t tensorDim, const Sharding &srcSharding, const Sharding &tgtSharding, ShapedType srcUnshardedType, TypedValue< ShapedType > srcShard) overridemlir::shard::SplitLastAxisPatterninlinevirtual
~ReshardingPattern()=defaultmlir::shard::ReshardingPatternvirtual