MLIR  22.0.0git
Public Member Functions | List of all members
mlir::shard::ElementwiseShardingInterface< ElemwiseOp > Struct Template Reference

#include "mlir/Dialect/Shard/Interfaces/ShardingInterfaceImpl.h"

+ Inheritance diagram for mlir::shard::ElementwiseShardingInterface< ElemwiseOp >:

Public Member Functions

SmallVector< utils::IteratorType > getLoopIteratorTypes (Operation *op) const
 
SmallVector< AffineMapgetIndexingMaps (Operation *op) const
 
LogicalResult partition (Operation *op, ArrayRef< Value > partitionedOperands, ArrayRef< Sharding > operandShardings, ArrayRef< Sharding > resultShardings, IRMapping &partitionMap, SymbolTableCollection &symbolTable, OpBuilder &builder) const
 

Detailed Description

template<typename ElemwiseOp>
struct mlir::shard::ElementwiseShardingInterface< ElemwiseOp >

Definition at line 105 of file ShardingInterfaceImpl.h.

Member Function Documentation

◆ getIndexingMaps()

template<typename ElemwiseOp >
SmallVector<AffineMap> mlir::shard::ElementwiseShardingInterface< ElemwiseOp >::getIndexingMaps ( Operation op) const
inline

◆ getLoopIteratorTypes()

template<typename ElemwiseOp >
SmallVector<utils::IteratorType> mlir::shard::ElementwiseShardingInterface< ElemwiseOp >::getLoopIteratorTypes ( Operation op) const
inline

◆ partition()

template<typename ElemwiseOp >
LogicalResult mlir::shard::ElementwiseShardingInterface< ElemwiseOp >::partition ( Operation op,
ArrayRef< Value partitionedOperands,
ArrayRef< Sharding operandShardings,
ArrayRef< Sharding resultShardings,
IRMapping partitionMap,
SymbolTableCollection symbolTable,
OpBuilder builder 
) const
inline

The documentation for this struct was generated from the following file: