Searched refs:revisedInputShape (Results 1 – 2 of 2) sorted by relevance
61 llvm::SmallVector<int64_t, 2> revisedInputShape{ in matchAndRewrite() local64 revisedInputShape, in matchAndRewrite()69 rewriter.getI64ArrayAttr(revisedInputShape)) in matchAndRewrite()
58 llvm::SmallVector<int64_t, 2> revisedInputShape{combined, inputShape[3]}; in matchAndRewrite() local60 RankedTensorType::get(revisedInputShape, inputType.getElementType()); in matchAndRewrite()64 rewriter.getI64ArrayAttr(revisedInputShape)) in matchAndRewrite()