Searched defs:rankDiff (Results 1 – 7 of 7) sorted by relevance
70 int64_t rankDiff = dstType.getRank() - srcType.getRank(); matchAndRewrite() local
350 int64_t rankDiff = in matchAndRewrite() local 1255 int rankDiff = srcType.getRank() - vectorType.getRank(); getTransferFoldableInnerUnitDims() local
89 Value rankDiff = std::get<1>(tup); in getBroadcastedDim() local286 Value shape, rankDiff; in matchAndRewrite() local
1673 int64_t rankDiff = broadcastSrcRank - extractResultRank; foldExtractFromBroadcast() local 2245 int64_t rankDiff = dstShape.size() - srcShape.size(); computeBroadcastedUnitDims() local 3051 unsigned rankDiff = destShape.size() - sourceShape.size(); verify() local 3742 unsigned rankDiff = dstRank - srcRank; matchAndRewrite() local [all...]
540 auto rankDiff = outShape.size() - shape.getRank(); resolveBroadcastShape() local
1107 int32_t rankDiff = dstRank - srcRank; vectorizeTensorExtract() local
2058 int rankDiff = inferredType.getRank() - desiredResultRank; inferCanonicalRankReducedResultType() local