Searched refs:originalOutputType (Results 1 – 2 of 2) sorted by relevance
77 auto originalOutputType = op->getOperand(1).getType().cast<MemRefType>(); in matchAsPerformingReduction() local83 if (!originalOutputType.hasStaticShape() || in matchAsPerformingReduction()84 originalOutputType.getNumElements() != 1) in matchAsPerformingReduction()114 auto originalOutputType = op->getOperand(1).getType().cast<MemRefType>(); in matchAndRewrite() local169 SmallVector<Value, 1> zeroIndices(originalOutputType.getRank(), zero); in matchAndRewrite()171 spirv::getElementPtr(*typeConverter, originalOutputType, convertedOutput, in matchAndRewrite()
411 auto originalOutputType = originalOutput.getType().cast<RankedTensorType>(); in splitReductionByScaling() local414 AffineMap map = b.getMultiDimIdentityMap(originalOutputType.getRank() + 1); in splitReductionByScaling()418 originalOutputType.getRank() + 1, getParallelIteratorTypeName()); in splitReductionByScaling()425 originalOutputType, in splitReductionByScaling()