Searched refs:revisedWeightShape (Results 1 – 2 of 2) sorted by relevance
73 llvm::SmallVector<int64_t, 2> revisedWeightShape{1, 1, 1, weightShape[2], in matchAndRewrite() local76 revisedWeightShape, in matchAndRewrite()81 rewriter.getI64ArrayAttr(revisedWeightShape)) in matchAndRewrite()
68 llvm::SmallVector<int64_t, 2> revisedWeightShape{weightShape[0], in matchAndRewrite() local71 revisedWeightShape, in matchAndRewrite()76 rewriter.getI64ArrayAttr(revisedWeightShape)) in matchAndRewrite()