Searched refs:reduceShape (Results 1 – 1 of 1) sorted by relevance
797 llvm::SmallVector<int64_t> reduceShape; in reduceMatchAndRewriteHelper() local801 reduceShape.push_back(inputTy.getDimSize(i)); in reduceMatchAndRewriteHelper()807 Type reduceTy = RankedTensorType::get(reduceShape, resultTy.getElementType()); in reduceMatchAndRewriteHelper()811 .create<linalg::InitTensorOp>(loc, dynDims, reduceShape, in reduceMatchAndRewriteHelper()