Searched defs:sourceShape (Results 1 – 9 of 9) sorted by relevance
338 assert(sourceShape.size() == readShape.size() && "expected same ranks."); in createReadOrMaskedRead() local
251 SmallVector<int64_t> targetShape = llvm::to_vector(sourceShape.getShape()); in allGatherResultShapeInUnsplitLastAxis() argument 377 allToAllResultShapeInMoveLastAxis(ShapedType sourceShape,int64_t splitCount,int64_t sourceTensorAxis,int64_t targetTensorAxis) allToAllResultShapeInMoveLastAxis() argument
32 mlir::getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape, in getReassociationIndicesForCollapse() argument [all...]
602 ArrayRef<int64_t> sourceShape = padOp.getSourceType().getShape(); matchAndRewrite() local
1578 ArrayRef<int64_t> sourceShape = unpackTensorType.getShape(); vectorizeAsTensorUnpackOp() local
2298 ArrayRef<int64_t> sourceShape = sourceType.getShape(); populateMap() local 2340 ArrayRef<int64_t> sourceShape = sourceType.getShape(); createNewOperandWithStaticSizes() local
2193 auto sourceShape = sourceTensorType.getShape(); rankReduceIfNeeded() local 3697 areAllInBound(ArrayRef<int64_t> sourceShape,ArrayRef<int64_t> limitShape) areAllInBound() argument 3970 getPackOpResultTypeShape(ArrayRef<int64_t> sourceShape,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getPackOpResultTypeShape() argument
3097 auto sourceShape = sourceMemrefType.getShape(); rankReduceIfNeeded() local 3134 ArrayRef<int64_t> sourceShape = subViewOp.getSourceType().getShape(); isTrivialSubViewOp() local
3032 auto sourceShape = sourceVectorType.getShape(); verify() local 3686 ArrayRef<int64_t> sourceShape = sourceVecTy.getShape(); matchAndRewrite() local 5206 auto sourceShape = sourceVectorType.getShape(); verifyVectorShapeCast() local [all...]