9 #ifndef MLIR_DIALECT_MESH_TRANSFORMS_SPMDIZATION_H
10 #define MLIR_DIALECT_MESH_TRANSFORMS_SPMDIZATION_H
38 TypedValue<ShapedType>
reshard(OpBuilder &builder,
MeshOp mesh, ShardOp source,
40 TypedValue<ShapedType> sourceShardValue);
41 TypedValue<ShapedType>
reshard(OpBuilder &builder, ShardOp source,
43 TypedValue<ShapedType> sourceShardValue,
44 SymbolTableCollection &symbolTableCollection);
TypedValue< ShapedType > reshard(OpBuilder &builder, MeshOp mesh, ShardOp source, ShardOp target, TypedValue< ShapedType > sourceShardValue)
void reshardingRegisterDependentDialects(DialectRegistry ®istry)
Include the generated interface declarations.