Spmdization.h File Reference

#include "mlir/Dialect/Mesh/IR/MeshOps.h"
#include "mlir/IR/DialectRegistry.h"

Go to the source code of this file.

Namespaces
	mlir
	Include the generated interface declarations.
	mlir::mesh

Functions
TypedValue< ShapedType >	mlir::mesh::reshard (OpBuilder &builder, MeshOp mesh, ShardOp source, ShardOp target, TypedValue< ShapedType > sourceShardValue)
TypedValue< ShapedType >	mlir::mesh::reshard (OpBuilder &builder, ShardOp source, ShardOp target, TypedValue< ShapedType > sourceShardValue, SymbolTableCollection &symbolTableCollection)
void	mlir::mesh::reshardingRegisterDependentDialects (DialectRegistry &registry)