Lines Matching refs:img2ColTensor
142 auto img2ColTensor = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
193 /*inputs=*/ValueRange{img2ColTensor.getResult(0), reshapedFilter}, in rewriteInIm2Col()
209 return std::make_pair(img2ColTensor.getOperation(), in rewriteInIm2Col()
314 auto img2ColTensor = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
336 loc, reshapedImg2ColTensorType, img2ColTensor.getResult(0), in rewriteInIm2Col()
360 return std::make_pair(img2ColTensor.getOperation(), in rewriteInIm2Col()
427 auto img2ColTensor = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
477 /*inputs=*/ValueRange{reshapedFilter, img2ColTensor.getResult(0)}, in rewriteInIm2Col()
493 return std::make_pair(img2ColTensor.getOperation(), in rewriteInIm2Col()
562 auto img2ColTensor = rewriter.create<linalg::GenericOp>( in rewriteInIm2Col() local
612 /*inputs=*/ValueRange{img2ColTensor.getResult(0), reshapedFilter}, in rewriteInIm2Col()
628 return std::make_pair(img2ColTensor.getOperation(), in rewriteInIm2Col()