Searched defs:colTensorShape (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConvertConv2DToImg2Col.cpp | 128 SmallVector<int64_t> colTensorShape = {n, oh * ow, fh * fw * ic}; in rewriteInIm2Col() local
288 SmallVector<int64_t> colTensorShape = {n, c, oh, ow, fh, fw}; in rewriteInIm2Col() local
414 SmallVector<int64_t, 4> colTensorShape = {n, ic * fh * fw, oh * ow}; in rewriteInIm2Col() local
548 SmallVector<int64_t> colTensorShape = {n, oh * ow, fh * fw * ic}; in rewriteInIm2Col() local
|