MLIR  20.0.0git
Public Attributes | List of all members
UnitExtentReplacementInfo Struct Reference

Compute the modified metadata for an operands of operation whose unit dims are being dropped. More...

Public Attributes

AffineMap indexMap
 
SmallVector< ReassociationIndicesreassociation
 
SmallVector< int64_t > targetShape
 

Detailed Description

Compute the modified metadata for an operands of operation whose unit dims are being dropped.

Return the new indexing map to use, the shape of the operand in the replacement op and the reassocation to use to go from original operand shape to modified operand shape.

Definition at line 331 of file DropUnitDims.cpp.

Member Data Documentation

◆ indexMap

AffineMap UnitExtentReplacementInfo::indexMap

Definition at line 332 of file DropUnitDims.cpp.

Referenced by dropUnitExtentFromOperandMetadata().

◆ reassociation

SmallVector<ReassociationIndices> UnitExtentReplacementInfo::reassociation

Definition at line 333 of file DropUnitDims.cpp.

Referenced by dropUnitExtentFromOperandMetadata().

◆ targetShape

SmallVector<int64_t> UnitExtentReplacementInfo::targetShape

Definition at line 334 of file DropUnitDims.cpp.

Referenced by dropUnitExtentFromOperandMetadata().


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