Searched defs:dstStaticShape (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 43 getCollapsedOutputDimFromInputShape(OpBuilder & builder,Location loc,int64_t dimIndex,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociationMap) getCollapsedOutputDimFromInputShape() argument
71 getCollapsedOutputShapeFromInputShape(OpBuilder & builder,Location loc,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation) getCollapsedOutputShapeFromInputShape() argument
83 getExpandedOutputDimFromInputShape(OpBuilder & builder,Location loc,int64_t dimIndex,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation,llvm::DenseMap<int64_t,int64_t> & expandedDimToCollapsedDim) getExpandedOutputDimFromInputShape() argument
123 getExpandedOutputShapeFromInputShape(OpBuilder & builder,Location loc,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation) getExpandedOutputShapeFromInputShape() argument
136 getReshapeOutputShapeFromInputShape(OpBuilder & builder,Location loc,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassocation) getReshapeOutputShapeFromInputShape() argument
|