MLIR
22.0.0git
|
#include "mlir/Dialect/Shard/Interfaces/ShardingInterfaceImpl.h"
Public Member Functions | |
SmallVector< utils::IteratorType > | getLoopIteratorTypes (Operation *op) const |
SmallVector< AffineMap > | getIndexingMaps (Operation *op) const |
LogicalResult | partition (Operation *op, ArrayRef< Value > partitionedOperands, ArrayRef< Sharding > operandShardings, ArrayRef< Sharding > resultShardings, IRMapping &partitionMap, SymbolTableCollection &symbolTable, OpBuilder &builder) const |
Definition at line 105 of file ShardingInterfaceImpl.h.
|
inline |
Definition at line 118 of file ShardingInterfaceImpl.h.
References mlir::Operation::getContext(), mlir::AffineMap::getMultiDimIdentityMap(), mlir::Operation::getNumOperands(), mlir::Operation::getNumResults(), mlir::Operation::getOperand(), and mlir::Value::getType().
|
inline |
Definition at line 108 of file ShardingInterfaceImpl.h.
References mlir::Operation::getOperand(), and mlir::Value::getType().
|
inline |
Definition at line 131 of file ShardingInterfaceImpl.h.
References mlir::shard::partitionTriviallyShardableOperation().