|
| bool | mlir::mesh::isReductionLoop (utils::IteratorType iType) |
| |
| template<typename T > |
| void | mlir::mesh::removeTrailingEmptySubArray (SmallVector< SmallVector< T >> &array) |
| |
| bool | mlir::mesh::isFullReplication (MeshSharding sharding) |
| |
| mesh::MeshOp | mlir::mesh::getMeshOrNull (Operation *op, FlatSymbolRefAttr meshSymbol, SymbolTableCollection &symbolTableCollection) |
| |
| mesh::MeshOp | mlir::mesh::getMesh (Operation *op, FlatSymbolRefAttr meshSymbol, SymbolTableCollection &symbolTableCollection) |
| |
| template<typename Op > |
| mesh::MeshOp | mlir::mesh::getMesh (Op op, SymbolTableCollection &symbolTableCollection) |
| |
| template<> |
| mesh::MeshOp | mlir::mesh::getMesh< ShardOp > (ShardOp op, SymbolTableCollection &symbolTableCollection) |
| |
| template<typename MeshAxesRange , typename MeshShapeRange > |
| int64_t | mlir::mesh::collectiveProcessGroupSize (MeshAxesRange &&meshAxes, MeshShapeRange &&meshShape) |
| |
| template<typename MeshAxesRange > |
| int64_t | mlir::mesh::collectiveProcessGroupSize (MeshAxesRange &&meshAxes, MeshOp mesh) |
| |
| int64_t | mlir::mesh::shardDimension (int64_t dimSize, int64_t shardCount) |
| |
| int64_t | mlir::mesh::gatherDimension (int64_t dimSize, int64_t shardCount) |
| |
| ShapedType | mlir::mesh::shardShapedType (ShapedType shape, MeshOp mesh, MeshSharding sharding) |
| |
| Type | mlir::mesh::shardType (Type type, MeshOp mesh, MeshSharding sharding) |
| |
| void | mlir::mesh::maybeInsertTargetShardingAnnotation (MeshSharding sharding, OpOperand &operand, OpBuilder &builder) |
| |
| void | mlir::mesh::maybeInsertTargetShardingAnnotation (MeshSharding sharding, OpResult result, OpBuilder &builder) |
| |
| void | mlir::mesh::maybeInsertSourceShardingAnnotation (MeshSharding sharding, OpOperand &operand, OpBuilder &builder) |
| |
