Searched defs:dstShape (Results 1 – 7 of 7) sorted by relevance
319 SmallVector<int64_t> dstShape = in matchAndRewrite() local 377 SmallVector<int64_t> dstShape; in matchAndRewrite() local
429 auto dstShape = insertOp.getDestVectorType().getShape(); in matchAndRewrite() local
154 auto dstShape = cast<TensorType>(dstType).getShape(); in createReassociationMapForCollapse() local [all...]
189 auto dstShape = dstTp.getShape(); in concatSizesFromInputs() local 997 ArrayRef<Size> dstShape = dstTp.getDimShape(); matchAndRewrite() local
908 ArrayRef<int64_t> dstShape = expandTy.getShape(); pushDownUnPackOpThroughExpandShape() local
220 OpBuilder &builder, Location loc, SmallVectorImpl<Value> &dstShape, in genReshapeDstShape() argument
2244 computeBroadcastedUnitDims(ArrayRef<int64_t> srcShape,ArrayRef<int64_t> dstShape) computeBroadcastedUnitDims() argument 2281 createOrFoldBroadcastOp(OpBuilder & b,Value value,ArrayRef<int64_t> dstShape,const llvm::SetVector<int64_t> & broadcastedDims) createOrFoldBroadcastOp() argument [all...]